flifloo/BetterTwitFix
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Changed variable and function naming conventions to match that of pythons spec

Browse source
This commit is contained in:
nbtm 2021-07-14 03:19:24 +10:00
parent c3c2dbb008
commit 59c4bbf188
1 changed files with 21 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
twitfix.py
View file

@ -84,19 +84,19 @@ def info(sub_path):
return result return result
def embed_video(vidlink): def embed_video(video_link):
cached_vnf = get_vnf_from_link_cache(vidlink) cached_vnf = get_vnf_from_link_cache(video_link)
if cached_vnf == None: if cached_vnf == None:
try: try:
vnf = link_to_vnf(vidlink) vnf = link_to_vnf(video_link)
add_vnf_to_link_cache(vidlink, vnf) add_vnf_to_link_cache(video_link, vnf)
return embed(vidlink, vnf) return embed(video_link, vnf)
except Exception as e: except Exception as e:
print(e) print(e)
return render_template('default.html', message="Failed to scan your link!") return render_template('default.html', message="Failed to scan your link!")
else: else:
return embed(vidlink, cached_vnf) return embed(video_link, cached_vnf)
def video_info(url, tweet="", desc="", thumb="", uploader=""): # Return a dict of video info with default values def video_info(url, tweet="", desc="", thumb="", uploader=""): # Return a dict of video info with default values
vnf = { vnf = {
@ -108,9 +108,9 @@ def video_info(url, tweet="", desc="", thumb="", uploader=""): # Return a dict o
} }
return vnf return vnf
def link_to_vnf_from_api(vidlink): def link_to_vnf_from_api(video_link):
print("Attempting to download tweet info from Twitter API") print("Attempting to download tweet info from Twitter API")
twid = int(re.sub(r'\?.*$','',vidlink.rsplit("/", 1)[-1])) # gets the tweet ID as a int from the passed url twid = int(re.sub(r'\?.*$','',video_link.rsplit("/", 1)[-1])) # gets the tweet ID as a int from the passed url
tweet = twitter_api.statuses.show(_id=twid, tweet_mode="extended") tweet = twitter_api.statuses.show(_id=twid, tweet_mode="extended")
if tweet['extended_entities']['media'][0]['video_info']['variants'][-1]['content_type'] == "video/mp4": if tweet['extended_entities']['media'][0]['video_info']['variants'][-1]['content_type'] == "video/mp4":
url = tweet['extended_entities']['media'][0]['video_info']['variants'][-1]['url'] url = tweet['extended_entities']['media'][0]['video_info']['variants'][-1]['url']
@ -125,34 +125,34 @@ def link_to_vnf_from_api(vidlink):
print(text) print(text)
print(len(text)) print(len(text))
vnf = video_info(url, vidlink, text, tweet['extended_entities']['media'][0]['media_url'], tweet['user']['name']) vnf = video_info(url, video_link, text, tweet['extended_entities']['media'][0]['media_url'], tweet['user']['name'])
return vnf return vnf
def link_to_vnf_from_youtubedl(vidlink): def link_to_vnf_from_youtubedl(video_link):
print("Attempting to download tweet info via YoutubeDL") print("Attempting to download tweet info via YoutubeDL")
with youtube_dl.YoutubeDL({'outtmpl': '%(id)s.%(ext)s'}) as ydl: with youtube_dl.YoutubeDL({'outtmpl': '%(id)s.%(ext)s'}) as ydl:
result = ydl.extract_info(vidlink, download=False) result = ydl.extract_info(video_link, download=False)
vnf = video_info(result['url'], vidlink, result['description'].rsplit(' ',1)[0], result['thumbnail'], result['uploader']) vnf = video_info(result['url'], video_link, result['description'].rsplit(' ',1)[0], result['thumbnail'], result['uploader'])
return vnf return vnf
def link_to_vnf(vidlink): # Return a VideoInfo object or die trying def link_to_vnf(video_link): # Return a VideoInfo object or die trying
if config['config']['method'] == 'hybrid': if config['config']['method'] == 'hybrid':
try: try:
return link_to_vnf_from_api(vidlink) return link_to_vnf_from_api(video_link)
except Exception as e: except Exception as e:
print("API Failed") print("API Failed")
print(e) print(e)
return link_to_vnf_from_youtubedl(vidlink) return link_to_vnf_from_youtubedl(video_link)
elif config['config']['method'] == 'api': elif config['config']['method'] == 'api':
try: try:
return link_to_vnf_from_api(vidlink) return link_to_vnf_from_api(video_link)
except Exception as e: except Exception as e:
print("API Failed") print("API Failed")
print(e) print(e)
return None return None
elif config['config']['method'] == 'youtube-dl': elif config['config']['method'] == 'youtube-dl':
try: try:
return link_to_vnf_from_youtubedl(vidlink) return link_to_vnf_from_youtubedl(video_link)
except Exception as e: except Exception as e:
print("Youtube-DL Failed") print("Youtube-DL Failed")
print(e) print(e)
@ -196,11 +196,11 @@ def add_vnf_to_link_cache(video_link, vnf):
json.dump(link_cache, outfile, indent=4, sort_keys=True) json.dump(link_cache, outfile, indent=4, sort_keys=True)
return None return None
def embed(vidlink, vnf): def embed(video_link, vnf):
desc = re.sub(r' http.*t\.co\S+', '', vnf['description'].replace("#","＃")) desc = re.sub(r' http.*t\.co\S+', '', vnf['description'].replace("#","＃"))
return render_template('index.html', vidurl=vnf['url'], desc=desc, pic=vnf['thumbnail'], user=vnf['uploader'], vidlink=vidlink) return render_template('index.html', vidurl=vnf['url'], desc=desc, pic=vnf['thumbnail'], user=vnf['uploader'], video_link=video_link)
def o_embed_gen(description, user, vidlink): def o_embed_gen(description, user, video_link):
out = { out = {
"type":"video", "type":"video",
"version":"1.0", "version":"1.0",
@ -208,7 +208,7 @@ def o_embed_gen(description, user, vidlink):
"provider_url":"https://github.com/robinuniverse/twitfix", "provider_url":"https://github.com/robinuniverse/twitfix",
"title":description, "title":description,
"author_name":user, "author_name":user,
"author_url":vidlink "author_url":video_link
} }
return out return out