muskit/NijiHolo_EN_ID_Bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

clean up create_ttweet_from_image()

Browse Source
This commit is contained in:
muskit
2023-08-20 16:13:38 -07:00
parent 9622929541
commit 75215e2e6d
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/util.py
+7 -7
View File
@@ -53,19 +53,19 @@ def get_key_from_value(d: dict, val):
# FIXME: web_auth_token under rate-limitation will fail to screenshot # FIXME: web_auth_token under rate-limitation will fail to screenshot
async def create_ttweet_image(ttweet): async def create_ttweet_image(ttweet):
tc = TweetCapture() tc = TweetCapture()
tc.cookies = [{'name': 'auth_token', 'value': dotenv_values()['web_auth_token']}] auth_token = dotenv_values().get('web_auth_token')
if auth_token:
tc.cookies = [{'name': 'auth_token', 'value': auth_token}]
if 'linux' in sys.platform: if 'linux' in sys.platform:
# Linux chromedriver path # Linux chromedriver path
tc.driver_path = '/usr/bin/chromedriver' tc.driver_path = '/usr/bin/chromedriver'
filename = f'{get_project_dir()}/img.png' filename = f'{get_project_dir()}/img.png'
url = ttweet.url()
img = None img = None
print(url)
try: os.remove(filename) try: os.remove(filename)
except: pass except: pass
try: try:
img = await tc.screenshot( img = await tc.screenshot(
url=url, url=ttweet.url(),
path=filename, path=filename,
mode=4, mode=4,
night_mode=1, night_mode=1,
@@ -76,9 +76,9 @@ async def create_ttweet_image(ttweet):
print('unable to create tweet image') print('unable to create tweet image')
traceback.print_exc() traceback.print_exc()
return None return None
else:
print(f'successfully saved {img}') print(f'successfully saved {img}')
return img return img
def get_tweet_url(id, username): def get_tweet_url(id, username):
return f'https://www.twitter.com/{username}/status/{id}' return f'https://www.twitter.com/{username}/status/{id}'