bradsbrown/wordlinator
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add db column to skip twitter, more verbose console logging

Browse Source
This commit is contained in:
Brad Brown
2022-06-02 08:34:20 -05:00
parent fcedc9b054
commit 60c212a803
3 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
wordlinator/app/__init__.py
View File

@@ -91,8 +91,10 @@ async def main_update(
if not any((s is not None for s in today_scores.values())): if not any((s is not None for s in today_scores.values())):
rich.print("[blue]No new scores found!") rich.print("[blue]No new scores found!")
else: else:
rich.print("[green]Updating scores in Sheets...")
updated_scores = sheets_client.update_scores(today_scores) updated_scores = sheets_client.update_scores(today_scores)
rich.print("[green]Saving scores in db...")
_save_db_scores(wordle_day, updated_scores) _save_db_scores(wordle_day, updated_scores)
print_score_table(wordle_day, today_scores) print_score_table(wordle_day, today_scores)

1
wordlinator/db/pg.py
View File

@@ -20,6 +20,7 @@ class User(BaseModel):
user_id = peewee.AutoField() user_id = peewee.AutoField()
username = peewee.CharField(unique=True) username = peewee.CharField(unique=True)
twitter_id = peewee.CharField(unique=True) twitter_id = peewee.CharField(unique=True)
check_twitter = peewee.BooleanField(default=True)
class Meta: class Meta:
table_name = "user_tbl" table_name = "user_tbl"

13
wordlinator/twitter/__init__.py
View File

@@ -125,9 +125,9 @@ class TwitterClient(httpx.AsyncClient):
return await self.get(self.USER_PATH.format(username=username)) return await self.get(self.USER_PATH.format(username=username))
async def get_user_id(self, username: str): async def get_user_id(self, username: str):
db_user_id = self.db.get_user_id(username) db_user = self.db.get_user(username)
if db_user_id: if db_user:
return db_user_id return db_user.twitter_id if db_user.check_twitter else False
else: else:
twitter_user = await self.get_user_by(username) twitter_user = await self.get_user_by(username)
user_id = None user_id = None
@@ -158,7 +158,7 @@ class TwitterClient(httpx.AsyncClient):
async def get_user_tweets_by(self, username: str): async def get_user_tweets_by(self, username: str):
user_id = await self.get_user_id(username) user_id = await self.get_user_id(username)
if not user_id: if not user_id:
return None return user_id
return await self.get_user_recent_tweets(user_id) return await self.get_user_recent_tweets(user_id)
async def get_user_wordles(self, username): async def get_user_wordles(self, username):
@@ -169,7 +169,10 @@ class TwitterClient(httpx.AsyncClient):
f"{user_tweets.status_code}: {user_tweets.text}" f"{user_tweets.status_code}: {user_tweets.text}"
) )
if not user_tweets: if not user_tweets:
rich.print(f"[yellow]No User ID found for {username}") if user_tweets is None:
rich.print(f"[yellow]No User ID found for {username}")
if user_tweets is False:
rich.print(f"[blue]Skipping check for {username}")
return [] return []
return self._build_wordle_tweets(user_tweets) return self._build_wordle_tweets(user_tweets)