Merge pull request #8 from aadi-1024/config

Config

Aaditya/config

@@ -0,0 +1,2 @@
+likes=50000
+num=1

Aaditya/config.py

@@ -0,0 +1,17 @@
+class Config:
+    def __init__(self, filename):
+        self.configpath = filename
+        self.config = {}
+        with open(self.configpath, "r") as file:
+            for i in file.readlines():
+                i = i.split("=")
+                self.config[i[0]] = i[1].strip()
+
+    def update(self, key, value):
+        self.config[key] = value
+        with open(self.configpath, 'w') as file:
+            for i, j in self.config.items():
+                file.write(f'{i}={j}\n')
+
+    def val(self, key):
+        return self.config[key]

Aaditya/main.py

@@ -1,11 +1,13 @@
 import discord
 from scrapper import Scraper
+from config import Config
 
 class Client(discord.Client):
 
     async def on_ready(self):
         print("Bot is online!")
         self.scraper = Scraper()
+        self.config = Config('config')
 
     async def on_message(self, message):
         if message.author == self.user:
@@ -19,7 +21,7 @@ async def on_message(self, message):
             await message.channel.send(f"Looking up twitter for {x}")
 
             try:
-                data = self.scraper.scrape(x, 5)
+                data = self.scraper.scrape(x, int(self.config.val('num')), self.config.val('likes'))
             except:
                 await message.channel.send(f"Couldn't find anything about {x} on twitter")
                 return
@@ -32,6 +34,23 @@ async def on_message(self, message):
 
             await message.channel.send(embed=discord.Embed(color=discord.Colour.blurple(), title=x, description=desc))
 
+        elif message.content.startswith('!setlikes'):
+            try:
+                x = int(message.content.split()[1])
+                self.config.update('likes', x)
+                await message.channel.send(f"Done! Minimum likes set to {x}")
+            except:
+                await message.channel.send("Please enter an integer value")
+
+        elif message.content.startswith('!setnum'):
+            try:
+
+                x = int(message.content.split()[1])
+                self.config.update('num', x)
+                await message.channel.send(f"Done! Number of entries set to {x}")
+            except:
+                await message.channel.send("Please enter an integer value")
+
 def main():
     #only works if command is invoked from the working directory itself
     with open(".env", "r") as file: 

Aaditya/scrapper.py

@@ -6,8 +6,8 @@
 class Scraper:
     def __init__(self):
         pass
-    def scrape(self, text, num):
-        fQuery=text+' min_faves:4000'
+    def scrape(self, text, num, likes):
+        fQuery=text+f' min_faves:{likes}'
         data=sntwitter.TwitterSearchScraper(fQuery)
         tweets=[]