Create followers.py

twint/followers.py

+from . import feed, get, db, output
+from bs4 import BeautifulSoup
+import aiohttp
+import asyncio
+import re
+import sys
+
+class Followers:
+	def __init__(self, config):
+		self.init = -1
+		self.feed = [-1]
+		self.count = 0
+		self.config = config
+
+		if self.config.Database:
+			print("Inserting into Database: " + str(self.config.Database))
+			self.conn = db.init(self.config.Database)
+			if isinstance(self.conn, str):
+				print(str)
+				sys.exit(1)
+
+		loop = asyncio.get_event_loop()
+		loop.run_until_complete(self.main())
+	
+	async def Feed(self):
+		ua = {'User-Agent': 'Lynx/2.8.5rel.1 libwww-FM/2.14 SSL-MM/1.4.1 GNUTLS/0.8.12'}
+		connect = aiohttp.TCPConnector(verify_ssl=False)
+		async with aiohttp.ClientSession(headers=ua, connector=connect) as session:
+			response = await get.Response(session, await get.Url(self.config, 
+				self.init).followers())
+		self.feed = []
+		try:
+			self.feed, self.init = feed.Follow(response)
+		except:
+			pass
+		return self.feed
+
+	async def follower(self):
+		await self.Feed()
+		for f in self.feed:
+			User = await output.getUser(f)
+			if self.config.Database:
+				db.followers(self.conn, self.config.Username, output)
+			
+			if self.config.Output != None:
+				output.write(User.name, self.config.Output)
+			
+			self.count += 1
+			print(User.name)
+	
+	async def main(self):
+		if self.config.User_id is not None:
+			self.config.Username = await get.Username(self.config.User_id)
+		while True:
+			if len(self.feed) > 0:
+				await self.follower()
+			else:
+				break
+
+			if self.config.Limit is not None and self.count >= int(self.config.Limit):
+				break
+
+		if self.config.Count:
+			print("[+] Finished: Successfully collected all {0.count} users who follow @{0.config.Username}.".format(self))