mcvsoftware/twblue
1
0
Fork 0
mirror of https://github.com/MCV-Software/TWBlue.git synced 2024-11-26 20:53:13 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

Improve save_users() and get_user() as those will be used in more places later

Browse Source
This commit is contained in:
Manuel Cortez 2021-06-25 13:11:33 -05:00
parent cdabd6f055
commit ab1a13f886
1 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/sessions/twitter/session.py
View File

@ -104,8 +104,6 @@ class Session(base.baseSession):
else: objects.insert(0, i) else: objects.insert(0, i)
incoming = incoming+1 incoming = incoming+1
pub.sendMessage("sent-dms-updated", total=sent, account=self.db["user_name"]) pub.sendMessage("sent-dms-updated", total=sent, account=self.db["user_name"])
self.db["direct_messages"] = objects
self.db["sent_direct_messages"] = sent_objects
return incoming return incoming
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
@ -387,7 +385,7 @@ class Session(base.baseSession):
""" Returns an user object associated with an ID. """ Returns an user object associated with an ID.
id str: User identifier, provided by Twitter. id str: User identifier, provided by Twitter.
returns a tweepy user object.""" returns a tweepy user object."""
if ("users" in self.db) == False or (id in self.db["users"]) == False: if ("users" in self.db) == False or (str(id) in self.db["users"]) == False:
try: try:
user = self.twitter.get_user(id=id) user = self.twitter.get_user(id=id)
except TweepError as err: except TweepError as err:
@ -395,9 +393,9 @@ class Session(base.baseSession):
user.screen_name = "deleted_user" user.screen_name = "deleted_user"
user.id = id user.id = id
user.name = _("Deleted account") user.name = _("Deleted account")
user.id_str = id return user
users = self.db["users"] users = self.db["users"]
users[user.id_str] = user users[user.id] = user
self.db["users"] = users self.db["users"] = users
return user return user
else: else:
@ -410,18 +408,18 @@ class Session(base.baseSession):
if ("users" in self.db) == False: if ("users" in self.db) == False:
user = utils.if_user_exists(self.twitter, screen_name) user = utils.if_user_exists(self.twitter, screen_name)
users = self.db["users"] users = self.db["users"]
users[user["id_str"]] = user users[user["id"]] = user
self.db["users"] = users self.db["users"] = users
return user["id_str"] return user["id"]
else: else:
for i in list(self.db["users"].keys()): for i in list(self.db["users"].keys()):
if self.db["users"][i].screen_name == screen_name: if self.db["users"][i].screen_name == screen_name:
return self.db["users"][i].id_str return self.db["users"][i].id
user = utils.if_user_exists(self.twitter, screen_name) user = utils.if_user_exists(self.twitter, screen_name)
users = self.db["users"] users = self.db["users"]
users[user.id_str] = user users[user.id] = user
self.db["users"] = users self.db["users"] = users
return user.id_str return user.id
def save_users(self, user_ids): def save_users(self, user_ids):
""" Adds all new users to the users database. """ """ Adds all new users to the users database. """