60 lines
2.3 KiB
Python
60 lines
2.3 KiB
Python
from flask import Flask, render_template
|
|
import youtube_dl
|
|
import json
|
|
import re
|
|
|
|
app = Flask(__name__)
|
|
pathregex = re.compile("\\w{1,15}\\/status\\/\\d{19}")
|
|
|
|
link_cache = {}
|
|
f = open('links.json',)
|
|
link_cache = json.load(f)
|
|
f.close()
|
|
|
|
@app.route('/<path:subpath>')
|
|
def twitfix(subpath):
|
|
|
|
match = pathregex.search(subpath)
|
|
if match is not None:
|
|
twitter_url = subpath
|
|
|
|
if match.start() == 0:
|
|
twitter_url = "https://twitter.com/" + subpath
|
|
|
|
if twitter_url in link_cache:
|
|
print("Link located in cache")
|
|
return render_template('index.html', vidurl=link_cache[twitter_url]['url'], tweet=link_cache[twitter_url]['description'], pic=link_cache[twitter_url]['thumbnail'], user=link_cache[twitter_url]['uploader'], tweeturl=twitter_url)
|
|
else:
|
|
with youtube_dl.YoutubeDL({'outtmpl': '%(id)s.%(ext)s'}) as ydl:
|
|
try:
|
|
print("Link not in cache, downloading and adding details to cache file")
|
|
result = ydl.extract_info(twitter_url, download=False)
|
|
|
|
link_cache[twitter_url] = {
|
|
"url" :result['url'],
|
|
"description" :result['description'],
|
|
"thumbnail" :result['thumbnail'],
|
|
"uploader" :result['uploader']
|
|
}
|
|
|
|
with open("links.json", "w") as outfile:
|
|
json.dump(link_cache, outfile, indent=4, sort_keys=True)
|
|
|
|
except Exception: # Just to keep from 500s that are messy
|
|
print(Exception)
|
|
return render_template('default.html', message="Failed to scan your twitter link! ( there is a chance I ran out of api calls.. wait a bit and try again )")
|
|
|
|
return render_template('index.html', vidurl=result['url'], tweet=result['description'], pic=result['thumbnail'], user=result['uploader'], tweeturl=twitter_url)
|
|
else:
|
|
print("Link invalid")
|
|
return render_template('default.html', message="Please use a twitter link!")
|
|
|
|
@app.route('/info/<path:subpath>')
|
|
def info(subpath):
|
|
with ydl:
|
|
result = ydl.extract_info(subpath, download=False)
|
|
|
|
return result
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host='0.0.0.0', port=80)
|