diff --git a/facebook/classes.py b/facebook/classes.py index 9534f7d..b69df6d 100644 --- a/facebook/classes.py +++ b/facebook/classes.py @@ -15,10 +15,11 @@ from helpers import sleep_, update_cookie class User: - def __init__(self, username): + def __init__(self, username, mode): self.collection = "members" self.username = str(username) - self.fetched = datetime.now().strftime("%Y%m%d_%H:%M:%S") + self.mode = mode + self.fetched = datetime.now().strftime("%Y%m%d") self.url_coverphotos = '' self.id = '' self.url_likes = '' @@ -28,7 +29,7 @@ class User: self.url = '' self.name = '' self.url_other_picture = '' - self.reacting_friends = 0 + self.reactions = 0 def add_to_db(self): # Lägg till profilen till arrango @@ -45,7 +46,8 @@ class User: "about": self.url_about, "cover photos": self.url_coverphotos, "fetched": self.fetched, - "reacting_friends": self.reacting_friends + "reactions": self.reactions, + 'mode': self.mode }, overwrite_mode="update", silent=True, @@ -57,10 +59,12 @@ class User: { "_id": "members/" + str(self.username), "checked": True, - "pictures_checked": self.profile_pictures, + "pictures_checked": self.profile_pictures, + "reaction": self.reactions }) + class Picture: def __init__(self, user): self.collection = "pictures" @@ -89,7 +93,14 @@ class Picture: ) class Profile: - def __init__(self, profile): + def __init__(self, profile, container): + """ Creates a new profile to do searches with. + + Args: + profile (dict): Document fetched from database. + container (string): Container name. + """ + # Uppdatera dokumentet i arango self.doc = profile self.doc['in_use'] = True @@ -102,6 +113,7 @@ class Profile: self.server = self.doc["server"] self.blocked = False + self.container = container # Ange proxies session = requests.Session() @@ -186,9 +198,8 @@ class Proxies: return self.proxies.pop(random.randrange(0, len(self.proxies), 1)) class Friend: - def __init__(self, user, mode): + def __init__(self, user): self.collection = "members" - self.mode = mode self.user = user # The friends friend self.username = '' self.url = ''