1073 lines
40 KiB
Python
1073 lines
40 KiB
Python
import datetime
|
|
import glob
|
|
import json
|
|
import math
|
|
import os
|
|
import random
|
|
import re
|
|
import time
|
|
import urllib
|
|
import subprocess
|
|
from multiprocessing import Process
|
|
from concurrent.futures import as_completed
|
|
|
|
import bleach
|
|
import feedparser
|
|
import requests
|
|
from bs4 import BeautifulSoup
|
|
from flask import Response
|
|
from flask import render_template, flash, redirect, url_for, request, send_from_directory, Markup
|
|
from flask_caching import Cache
|
|
from flask_login import login_user, logout_user, current_user, login_required
|
|
from numerize import numerize
|
|
from requests_futures.sessions import FuturesSession
|
|
from werkzeug.datastructures import Headers
|
|
from werkzeug.urls import url_parse
|
|
from werkzeug.utils import secure_filename
|
|
from youtube_search import YoutubeSearch
|
|
|
|
from app import app, db
|
|
from app.forms import LoginForm, RegistrationForm, EmptyForm, SearchForm, ChannelForm
|
|
from app.models import User, twitterPost, ytPost, Post, youtubeFollow, twitterFollow
|
|
|
|
from youtube import comments, utils, channel as ytch, search as yts
|
|
from youtube import watch as ytwatch
|
|
from youtube import video as ytvid
|
|
|
|
from nitter import feed as nitterfeed
|
|
from nitter import user as nitteruser
|
|
|
|
#########################################
|
|
|
|
#########################################
|
|
|
|
cache = Cache(config={'CACHE_TYPE': 'simple'})
|
|
cache.init_app(app)
|
|
##########################
|
|
#### Config variables ####
|
|
##########################
|
|
config = json.load(open('yotter-config.json'))
|
|
##########################
|
|
#### Config variables ####
|
|
##########################
|
|
NITTERINSTANCE = config['nitterInstance'] # Must be https://.../
|
|
YOUTUBERSS = "https://www.youtube.com/feeds/videos.xml?channel_id="
|
|
|
|
|
|
##########################
|
|
#### Global variables ####
|
|
##########################
|
|
|
|
#########################
|
|
#### Twitter Logic ######
|
|
#########################
|
|
@app.before_request
|
|
def before_request():
|
|
if current_user.is_authenticated:
|
|
current_user.set_last_seen()
|
|
db.session.commit()
|
|
|
|
|
|
@app.route('/')
|
|
@app.route('/index')
|
|
@login_required
|
|
@cache.cached(timeout=50, key_prefix='home')
|
|
def index():
|
|
return render_template('home.html', config=config)
|
|
|
|
|
|
@app.route('/twitter')
|
|
@app.route('/twitter/<page>')
|
|
@login_required
|
|
def twitter(page=0):
|
|
followingList = current_user.twitter_following_list()
|
|
form = EmptyForm()
|
|
followCount = len(followingList)
|
|
page = int(page)
|
|
avatarPath = "img/avatars/1.png"
|
|
|
|
followList = []
|
|
for f in followingList:
|
|
followList.append(f.username)
|
|
posts = []
|
|
|
|
cache_file = glob.glob("app/cache/{}_*".format(current_user.username))
|
|
if (len(cache_file) > 0):
|
|
time_diff = round(time.time() - os.path.getmtime(cache_file[0]))
|
|
else:
|
|
time_diff = 999
|
|
|
|
# If cache file is more than 1 minute old
|
|
if page == 0 and time_diff > 60:
|
|
if cache_file:
|
|
for f in cache_file:
|
|
os.remove(f)
|
|
feed = nitterfeed.get_feed(followList)
|
|
cache_file = "{u}_{d}.json".format(u=current_user.username, d=time.strftime("%Y%m%d-%H%M%S"))
|
|
with open("app/cache/{}".format(cache_file), 'w') as fp:
|
|
json.dump(feed, fp)
|
|
|
|
# Else, refresh feed
|
|
else:
|
|
try:
|
|
cache_file = glob.glob("app/cache/{}*".format(current_user.username))[0]
|
|
with open(cache_file, 'r') as fp:
|
|
feed = json.load(fp)
|
|
except:
|
|
feed = nitterfeed.get_feed(followList)
|
|
cache_file = "{u}_{d}.json".format(u=current_user.username, d=time.strftime("%Y%m%d-%H%M%S"))
|
|
with open("app/cache/{}".format(cache_file), 'w') as fp:
|
|
json.dump(feed, fp)
|
|
|
|
posts.extend(feed)
|
|
# Items range per page
|
|
page_items = page * 16
|
|
offset = page_items + 16
|
|
# Pagination logic
|
|
init_page = page - 3
|
|
if init_page < 0:
|
|
init_page = 0
|
|
total_pages = page + 5
|
|
max_pages = int(math.ceil(len(posts) / 10)) # Total number of pages.
|
|
if total_pages > max_pages:
|
|
total_pages = max_pages
|
|
|
|
# Posts to render
|
|
if posts and len(posts) > offset:
|
|
posts = posts[page_items:offset]
|
|
else:
|
|
posts = posts[page_items:]
|
|
return render_template('twitter.html', title='Yotter | Twitter', posts=posts, followedCount=followCount, form=form, config=config,
|
|
pages=total_pages, init_page=init_page, actual_page=page)
|
|
|
|
|
|
@app.route('/savePost/<url>', methods=['POST'])
|
|
@login_required
|
|
def savePost(url):
|
|
savedUrl = url.replace('~', '/')
|
|
r = requests.get(savedUrl)
|
|
html = BeautifulSoup(str(r.content), "lxml")
|
|
post = html.body.find('div', attrs={'class': 'main-tweet'})
|
|
|
|
newPost = Post()
|
|
newPost.url = savedUrl
|
|
newPost.username = post.find('a', 'username').text.replace("@", "")
|
|
newPost.body = post.find_all('div', attrs={'class': 'tweet-content'})[0].text.encode('latin1').decode(
|
|
'unicode_escape').encode('latin1').decode('utf8')
|
|
newPost.timestamp = post.find_all('p', attrs={'class': 'tweet-published'})[0].text.encode('latin1').decode(
|
|
'unicode_escape').encode('latin1').decode('utf8')
|
|
newPost.user_id = current_user.id
|
|
try:
|
|
db.session.add(newPost)
|
|
db.session.commit()
|
|
except:
|
|
flash("Post could not be saved. Either it was already saved or there was an error.")
|
|
return redirect(request.referrer)
|
|
|
|
|
|
@app.route('/saved')
|
|
@login_required
|
|
def saved():
|
|
savedPosts = current_user.saved_posts().all()
|
|
return render_template('saved.html', title='Saved', savedPosts=savedPosts, config=config)
|
|
|
|
|
|
@app.route('/deleteSaved/<id>', methods=['POST'])
|
|
@login_required
|
|
def deleteSaved(id):
|
|
savedPost = Post.query.filter_by(id=id).first()
|
|
db.session.delete(savedPost)
|
|
db.session.commit()
|
|
return redirect(url_for('saved'))
|
|
|
|
|
|
@app.route('/follow/<username>', methods=['POST'])
|
|
@login_required
|
|
def follow(username):
|
|
form = EmptyForm()
|
|
if form.validate_on_submit():
|
|
if followTwitterAccount(username):
|
|
flash("{} followed!".format(username))
|
|
return redirect(request.referrer)
|
|
|
|
|
|
def followTwitterAccount(username):
|
|
if isTwitterUser(username):
|
|
if not current_user.is_following_tw(username):
|
|
try:
|
|
follow = twitterFollow()
|
|
follow.username = username
|
|
follow.followers.append(current_user)
|
|
db.session.add(follow)
|
|
db.session.commit()
|
|
return True
|
|
except:
|
|
flash("Twitter: Couldn't follow {}. Already followed?".format(username))
|
|
return False
|
|
else:
|
|
flash("Something went wrong... try again")
|
|
return False
|
|
|
|
|
|
@app.route('/unfollow/<username>', methods=['POST'])
|
|
@login_required
|
|
def unfollow(username):
|
|
form = EmptyForm()
|
|
if form.validate_on_submit():
|
|
if twUnfollow(username):
|
|
flash("{} unfollowed!".format(username))
|
|
return redirect(request.referrer)
|
|
|
|
|
|
def twUnfollow(username):
|
|
try:
|
|
user = twitterFollow.query.filter_by(username=username).first()
|
|
db.session.delete(user)
|
|
db.session.commit()
|
|
except:
|
|
flash("There was an error unfollowing the user. Try again.")
|
|
return redirect(request.referrer)
|
|
|
|
|
|
@app.route('/following')
|
|
@login_required
|
|
def following():
|
|
form = EmptyForm()
|
|
followCount = len(current_user.twitter_following_list())
|
|
accounts = current_user.twitter_following_list()
|
|
return render_template('following.html', accounts=accounts, count=followCount, form=form, config=config)
|
|
|
|
|
|
@app.route('/search', methods=['GET', 'POST'])
|
|
@login_required
|
|
def search():
|
|
form = SearchForm()
|
|
if form.validate_on_submit():
|
|
user = form.username.data
|
|
results = twitterUserSearch(user)
|
|
if results:
|
|
return render_template('search.html', form=form, results=results, config=config)
|
|
else:
|
|
flash("User {} not found...".format(user))
|
|
return redirect(request.referrer)
|
|
else:
|
|
return render_template('search.html', form=form, config=config)
|
|
|
|
@app.route('/u/<username>')
|
|
@app.route('/<username>')
|
|
@app.route('/<username>/<page>')
|
|
@login_required
|
|
def u(username, page=1):
|
|
page=int(page)
|
|
if username == "favicon.ico":
|
|
return redirect(url_for('static', filename='favicons/favicon.ico'))
|
|
form = EmptyForm()
|
|
avatarPath = "img/avatars/{}.png".format(str(random.randint(1, 12)))
|
|
user = nitteruser.get_user_info(username)
|
|
if not user:
|
|
flash("This user is not on Twitter.")
|
|
return redirect(request.referrer)
|
|
|
|
posts = []
|
|
tweets=nitteruser.get_tweets(username, page)
|
|
if tweets == 'Empty feed':
|
|
posts = False
|
|
elif tweets == 'Protected feed':
|
|
posts = 'Protected'
|
|
else:
|
|
posts.extend(tweets)
|
|
|
|
if page-1 < 0:
|
|
prev_page = 0
|
|
else:
|
|
prev_page = page-1
|
|
|
|
if page > 2:
|
|
page =2
|
|
|
|
return render_template('user.html', posts=posts, user=user, form=form, config=config, page=page, prev_page=prev_page)
|
|
|
|
|
|
#########################
|
|
#### Youtube Logic ######
|
|
#########################
|
|
@app.route('/youtube', methods=['GET', 'POST'])
|
|
@login_required
|
|
def youtube():
|
|
followCount = len(current_user.youtube_following_list())
|
|
start_time = time.time()
|
|
ids = current_user.youtube_following_list()
|
|
videos = getYoutubePosts(ids)
|
|
if videos:
|
|
videos.sort(key=lambda x: x.date, reverse=True)
|
|
print("--- {} seconds fetching youtube feed---".format(time.time() - start_time))
|
|
return render_template('youtube.html', title="Yotter | Youtube", videos=videos, followCount=followCount,
|
|
config=config)
|
|
|
|
|
|
@app.route('/ytfollowing', methods=['GET', 'POST'])
|
|
@login_required
|
|
def ytfollowing():
|
|
form = EmptyForm()
|
|
channelList = current_user.youtube_following_list()
|
|
channelCount = len(channelList)
|
|
|
|
return render_template('ytfollowing.html', form=form, channelList=channelList, channelCount=channelCount,
|
|
config=config)
|
|
|
|
|
|
@app.route('/ytsearch', methods=['GET', 'POST'])
|
|
@login_required
|
|
def ytsearch():
|
|
form = ChannelForm()
|
|
button_form = EmptyForm()
|
|
query = request.args.get('q', None)
|
|
sort = request.args.get('s', None)
|
|
if sort != None:
|
|
sort = int(sort)
|
|
else:
|
|
sort = 0
|
|
|
|
page = request.args.get('p', None)
|
|
if page == None:
|
|
page = 1
|
|
|
|
if query:
|
|
autocorrect = 1
|
|
filters = {"time": 0, "type": 0, "duration": 0}
|
|
results = yts.search_by_terms(query, page, autocorrect, sort, filters)
|
|
|
|
next_page = "/ytsearch?q={q}&s={s}&p={p}".format(q=query, s=sort, p=int(page) + 1)
|
|
if int(page) == 1:
|
|
prev_page = "/ytsearch?q={q}&s={s}&p={p}".format(q=query, s=sort, p=1)
|
|
else:
|
|
prev_page = "/ytsearch?q={q}&s={s}&p={p}".format(q=query, s=sort, p=int(page) - 1)
|
|
|
|
for video in results['videos']:
|
|
hostname = urllib.parse.urlparse(video['videoThumb']).netloc
|
|
video['videoThumb'] = video['videoThumb'].replace("https://{}".format(hostname), "") + "&host=" + hostname
|
|
|
|
for channel in results['channels']:
|
|
if config['isInstance']:
|
|
channel['thumbnail'] = channel['thumbnail'].replace("~", "/")
|
|
hostName = urllib.parse.urlparse(channel['thumbnail']).netloc
|
|
channel['thumbnail'] = channel['thumbnail'].replace("https://{}".format(hostName),
|
|
"") + "?host=" + hostName
|
|
return render_template('ytsearch.html', form=form, btform=button_form, results=results,
|
|
restricted=config['restrictPublicUsage'], config=config, npage=next_page,
|
|
ppage=prev_page)
|
|
else:
|
|
return render_template('ytsearch.html', form=form, results=False)
|
|
|
|
|
|
@app.route('/ytfollow/<channelId>', methods=['POST'])
|
|
@login_required
|
|
def ytfollow(channelId):
|
|
r = followYoutubeChannel(channelId)
|
|
return redirect(request.referrer)
|
|
|
|
|
|
def followYoutubeChannel(channelId):
|
|
try:
|
|
try:
|
|
if not current_user.is_following_yt(channelId):
|
|
channelData = ytch.get_channel_tab_info(channelId, tab='about')
|
|
if channelData == False:
|
|
return False
|
|
follow = youtubeFollow()
|
|
follow.channelId = channelId
|
|
follow.channelName = channelData['channel_name']
|
|
follow.followers.append(current_user)
|
|
db.session.add(follow)
|
|
db.session.commit()
|
|
flash("{} followed!".format(channelData['channel_name']))
|
|
return True
|
|
else:
|
|
return False
|
|
except Exception as e:
|
|
print(e)
|
|
return False
|
|
except KeyError as ke:
|
|
print("KeyError: {}:'{}' could not be found".format(ke, channelId))
|
|
flash("Youtube: ChannelId '{}' is not valid".format(channelId))
|
|
return False
|
|
|
|
|
|
@app.route('/ytunfollow/<channelId>', methods=['POST'])
|
|
@login_required
|
|
def ytunfollow(channelId):
|
|
unfollowYoutubeChannel(channelId)
|
|
return redirect(request.referrer)
|
|
|
|
|
|
def unfollowYoutubeChannel(channelId):
|
|
try:
|
|
channel = youtubeFollow.query.filter_by(channelId=channelId).first()
|
|
name = channel.channelName
|
|
db.session.delete(channel)
|
|
db.session.commit()
|
|
channel = youtubeFollow.query.filter_by(channelId=channelId).first()
|
|
if channel:
|
|
db.session.delete(channel)
|
|
db.session.commit()
|
|
flash("{} unfollowed!".format(name))
|
|
except:
|
|
flash("There was an error unfollowing the user. Try again.")
|
|
|
|
|
|
@app.route('/channel/<id>', methods=['GET'])
|
|
@app.route('/user/<id>', methods=['GET'])
|
|
@app.route('/c/<id>', methods=['GET'])
|
|
@login_required
|
|
def channel(id):
|
|
form = ChannelForm()
|
|
button_form = EmptyForm()
|
|
|
|
page = request.args.get('p', None)
|
|
sort = request.args.get('s', None)
|
|
if page is None:
|
|
page = 1
|
|
if sort is None:
|
|
sort = 3
|
|
|
|
data = ytch.get_channel_tab_info(id, page, sort)
|
|
|
|
for video in data['items']:
|
|
if config['isInstance']:
|
|
hostName = urllib.parse.urlparse(video['thumbnail'][1:]).netloc
|
|
video['thumbnail'] = video['thumbnail'].replace("https://{}".format(hostName), "")[1:].replace("hqdefault",
|
|
"mqdefault") + "&host=" + hostName
|
|
else:
|
|
video['thumbnail'] = video['thumbnail'].replace('/', '~')
|
|
|
|
if config['isInstance']:
|
|
hostName = urllib.parse.urlparse(data['avatar'][1:]).netloc
|
|
data['avatar'] = data['avatar'].replace("https://{}".format(hostName), "")[1:] + "?host=" + hostName
|
|
else:
|
|
data['avatar'] = data['avatar'].replace('/', '~')
|
|
|
|
next_page = "/channel/{q}?s={s}&p={p}".format(q=id, s=sort, p=int(page) + 1)
|
|
if int(page) == 1:
|
|
prev_page = "/channel/{q}?s={s}&p={p}".format(q=id, s=sort, p=1)
|
|
else:
|
|
prev_page = "/channel/{q}?s={s}&p={p}".format(q=id, s=sort, p=int(page) - 1)
|
|
|
|
return render_template('channel.html', form=form, btform=button_form, data=data,
|
|
restricted=config['restrictPublicUsage'], config=config, next_page=next_page,
|
|
prev_page=prev_page)
|
|
|
|
|
|
def get_best_urls(urls):
|
|
'''Gets URLS in youtube format (format_id, url, height) and returns best ones for yotter'''
|
|
best_formats = ["22", "18", "34", "35", "36", "37", "38", "43", "44", "45", "46"]
|
|
best_urls = []
|
|
for url in urls:
|
|
for f in best_formats:
|
|
if url['format_id'] == f:
|
|
best_urls.append(url)
|
|
return best_urls
|
|
|
|
|
|
def get_live_urls(urls):
|
|
"""Gets URLS in youtube format (format_id, url, height) and returns best ones for yotter"""
|
|
best_formats = ["91", "92", "93", "94", "95", "96"]
|
|
best_urls = []
|
|
for url in urls:
|
|
for f in best_formats:
|
|
if url['format_id'] == f:
|
|
best_urls.append(url)
|
|
return best_urls
|
|
|
|
|
|
@app.route('/watch', methods=['GET'])
|
|
@login_required
|
|
def watch():
|
|
id = request.args.get('v', None)
|
|
info = ytvid.get_info(id)
|
|
|
|
if info['error'] == False:
|
|
for format in info['formats']:
|
|
hostName = urllib.parse.urlparse(format['url']).netloc
|
|
format['url'] = format['url'].replace("https://{}".format(hostName), "") + "&host=" + hostName
|
|
|
|
for format in info['audio_formats']:
|
|
hostName = urllib.parse.urlparse(format['url']).netloc
|
|
format['url'] = format['url'].replace("https://{}".format(hostName), "") + "&host=" + hostName
|
|
|
|
# Markup description
|
|
try:
|
|
info['description'] = Markup(bleach.linkify(info['description'].replace("\n", "<br>"))).replace(
|
|
'www.youtube.com', config['serverName']).replace('youtube.com', config['serverName']).replace("/join",
|
|
"")
|
|
except AttributeError or TypeError:
|
|
print(info['description'])
|
|
|
|
# Get comments
|
|
if not info['is_live']:
|
|
videocomments = comments.video_comments(id, sort=0, offset=0, lc='', secret_key='')
|
|
videocomments = utils.post_process_comments_info(videocomments)
|
|
if videocomments is not None:
|
|
videocomments.sort(key=lambda x: x['likes'], reverse=True)
|
|
else:
|
|
videocomments = False
|
|
|
|
return render_template("video.html", info=info, title=info['title'], config=config,
|
|
videocomments=videocomments)
|
|
|
|
return render_template("video.html", info=info, title='Scheduled Video', config=config)
|
|
|
|
|
|
def markupString(string):
|
|
string = string.replace("\n\n", "<br><br>").replace("\n", "<br>")
|
|
string = bleach.linkify(string)
|
|
string = string.replace("https://youtube.com/", "")
|
|
string = string.replace("https://www.youtube.com/", "")
|
|
string = string.replace("https://twitter.com/", "/u/")
|
|
return Markup(string)
|
|
|
|
|
|
## PROXY videos through Yotter server to the client.
|
|
@app.route('/stream/<url>', methods=['GET', 'POST'])
|
|
@login_required
|
|
def stream(url):
|
|
# This function proxies the video stream from GoogleVideo to the client.
|
|
url = url.replace('YotterSlash', '/')
|
|
headers = Headers()
|
|
if (url):
|
|
s = requests.Session()
|
|
s.verify = True
|
|
req = s.get(url, stream=True)
|
|
headers.add('Range', request.headers['Range'])
|
|
headers.add('Accept-Ranges', 'bytes')
|
|
headers.add('Content-Length', str(int(req.headers['Content-Length']) + 1))
|
|
response = Response(req.iter_content(chunk_size=10 * 1024), mimetype=req.headers['Content-Type'],
|
|
content_type=req.headers['Content-Type'], direct_passthrough=True, headers=headers)
|
|
# enable browser file caching with etags
|
|
response.cache_control.public = True
|
|
response.cache_control.max_age = int(60000)
|
|
return response
|
|
else:
|
|
flash("Something went wrong loading the video... Try again.")
|
|
return redirect(url_for('youtube'))
|
|
|
|
|
|
def download_file(streamable):
|
|
with streamable as stream:
|
|
stream.raise_for_status()
|
|
for chunk in stream.iter_content(chunk_size=8192):
|
|
yield chunk
|
|
|
|
|
|
#########################
|
|
#### General Logic ######
|
|
#########################
|
|
@app.route('/login', methods=['GET', 'POST'])
|
|
def login():
|
|
if current_user.is_authenticated:
|
|
return redirect(url_for('index'))
|
|
form = LoginForm()
|
|
if form.validate_on_submit():
|
|
user = User.query.filter_by(username=form.username.data).first()
|
|
if user is None or not user.check_password(form.password.data):
|
|
flash('Invalid username or password')
|
|
return redirect(url_for('login'))
|
|
if user.username == config['admin_user']:
|
|
user.set_admin_user()
|
|
db.session.commit()
|
|
login_user(user, remember=form.remember_me.data)
|
|
next_page = request.args.get('next')
|
|
if not next_page or url_parse(next_page).netloc != '':
|
|
next_page = url_for('index')
|
|
return redirect(next_page)
|
|
return render_template('login.html', title='Sign In', form=form, config=config)
|
|
|
|
|
|
# Proxy images through server
|
|
@app.route('/img/<url>', methods=['GET', 'POST'])
|
|
@login_required
|
|
def img(url):
|
|
pic = requests.get(url.replace("~", "/"))
|
|
return Response(pic, mimetype="image/png")
|
|
|
|
|
|
@app.route('/logout')
|
|
def logout():
|
|
logout_user()
|
|
return redirect(url_for('index'))
|
|
|
|
|
|
@app.route('/settings')
|
|
@login_required
|
|
@cache.cached(timeout=50, key_prefix='settings')
|
|
def settings():
|
|
active = 0
|
|
users = db.session.query(User).all()
|
|
for u in users:
|
|
if u.last_seen == None:
|
|
u.set_last_seen()
|
|
db.session.commit()
|
|
else:
|
|
t = datetime.datetime.utcnow() - u.last_seen
|
|
s = t.total_seconds()
|
|
m = s / 60
|
|
if m < 25:
|
|
active = active + 1
|
|
|
|
instanceInfo = {
|
|
"totalUsers": db.session.query(User).count(),
|
|
"active": active,
|
|
}
|
|
return render_template('settings.html', info=instanceInfo, config=config, admin=current_user.is_admin)
|
|
|
|
|
|
'''@app.route('/clear_inactive_users/<phash>')
|
|
@login_required
|
|
def clear_inactive_users(phash):
|
|
ahash = User.query.filter_by(username=config['admin_user']).first().password_hash
|
|
if phash == ahash:
|
|
users = db.session.query(User).all()
|
|
for u in users:
|
|
if u.username == config['admin_user']:
|
|
continue
|
|
t = datetime.datetime.utcnow() - u.last_seen
|
|
t = math.floor(t.total_seconds())
|
|
max_old_s = config['max_old_user_days']*86400
|
|
if t > max_old_s:
|
|
user = User.query.filter_by(username=u.username).first()
|
|
print("deleted "+u.username)
|
|
db.session.delete(user)
|
|
db.session.commit()
|
|
else:
|
|
flash("You must be admin for this action")
|
|
return redirect(request.referrer)'''
|
|
|
|
|
|
@app.route('/export')
|
|
@login_required
|
|
# Export data into a JSON file. Later you can import the data.
|
|
def export():
|
|
a = exportData()
|
|
if a:
|
|
return send_from_directory('.', 'data_export.json', as_attachment=True)
|
|
else:
|
|
return redirect(url_for('error/405'))
|
|
|
|
|
|
def exportData():
|
|
twitterFollowing = current_user.twitter_following_list()
|
|
youtubeFollowing = current_user.youtube_following_list()
|
|
data = {}
|
|
data['twitter'] = []
|
|
data['youtube'] = []
|
|
|
|
for f in twitterFollowing:
|
|
data['twitter'].append({
|
|
'username': f.username
|
|
})
|
|
|
|
for f in youtubeFollowing:
|
|
data['youtube'].append({
|
|
'channelId': f.channelId
|
|
})
|
|
|
|
try:
|
|
with open('app/data_export.json', 'w') as outfile:
|
|
json.dump(data, outfile)
|
|
return True
|
|
except:
|
|
return False
|
|
|
|
|
|
@app.route('/importdata', methods=['GET', 'POST'])
|
|
@login_required
|
|
def importdata():
|
|
if request.method == 'POST':
|
|
# check if the post request has the file part
|
|
if 'file' not in request.files:
|
|
flash('No file part')
|
|
return redirect(request.referrer)
|
|
file = request.files['file']
|
|
# if user does not select file, browser also
|
|
# submit an empty part without filename
|
|
if file.filename == '':
|
|
flash('No selected file')
|
|
return redirect(request.referrer)
|
|
else:
|
|
flash("Data is being imported. You can keep using Yotter.")
|
|
importdataasync(file)
|
|
return redirect(request.referrer)
|
|
|
|
return redirect(request.referrer)
|
|
|
|
|
|
def importdataasync(file):
|
|
p = Process(target=importdataforprocess, args=(file,))
|
|
p.start()
|
|
|
|
|
|
def importdataforprocess(file):
|
|
option = request.form['import_format']
|
|
if option == 'yotter':
|
|
importYotterSubscriptions(file)
|
|
elif option == 'youtube':
|
|
importYoutubeSubscriptions(file)
|
|
|
|
|
|
@app.route('/deleteme', methods=['GET', 'POST'])
|
|
@login_required
|
|
def deleteme():
|
|
user = User.query.filter_by(username=current_user.username).first()
|
|
db.session.delete(user)
|
|
db.session.commit()
|
|
logout_user()
|
|
return redirect(url_for('index'))
|
|
|
|
|
|
def importYoutubeSubscriptions(file):
|
|
filename = secure_filename(file.filename)
|
|
try:
|
|
data = re.findall('(UC[a-zA-Z0-9_-]{22})|(?<=user/)[a-zA-Z0-9_-]+', file.read().decode('utf-8'))
|
|
for acc in data:
|
|
r = followYoutubeChannel(acc)
|
|
except Exception as e:
|
|
print(e)
|
|
flash("File is not valid.")
|
|
|
|
|
|
def importYotterSubscriptions(file):
|
|
filename = secure_filename(file.filename)
|
|
data = json.load(file)
|
|
for acc in data['twitter']:
|
|
r = followTwitterAccount(acc['username'])
|
|
|
|
for acc in data['youtube']:
|
|
r = followYoutubeChannel(acc['channelId'])
|
|
|
|
|
|
@app.route('/register', methods=['GET', 'POST'])
|
|
def register():
|
|
form = RegistrationForm()
|
|
if current_user.is_authenticated:
|
|
return redirect(url_for('index'))
|
|
|
|
REGISTRATIONS = True
|
|
try:
|
|
count = db.session.query(User).count()
|
|
if count >= config['maxInstanceUsers'] or config['maxInstanceUsers'] == 0:
|
|
REGISTRATIONS = False
|
|
except:
|
|
REGISTRATIONS = True
|
|
|
|
if form.validate_on_submit():
|
|
if User.query.filter_by(username=form.username.data).first():
|
|
flash("This username is taken! Try with another.")
|
|
return redirect(request.referrer)
|
|
|
|
user = User(username=form.username.data)
|
|
user.set_password(form.password.data)
|
|
db.session.add(user)
|
|
db.session.commit()
|
|
flash('Congratulations, you are now a registered user!')
|
|
return redirect(url_for('login'))
|
|
|
|
return render_template('register.html', title='Register', registrations=REGISTRATIONS, form=form, config=config)
|
|
|
|
|
|
@app.route('/status')
|
|
def status():
|
|
count = db.session.query(User).count()
|
|
if count >= config['maxInstanceUsers'] or config['maxInstanceUsers'] == 0:
|
|
filen = url_for('static', filename='img/close.png')
|
|
caniregister = False
|
|
else:
|
|
filen = url_for('static', filename='img/open.png')
|
|
caniregister = True
|
|
|
|
try:
|
|
with open ("version.txt", "r") as versionFile:
|
|
ver=versionFile.readlines()
|
|
hsh = ver[0]
|
|
update= ver[1]
|
|
tag=ver[2]
|
|
except:
|
|
try:
|
|
tag = str(subprocess.check_output(["git", "describe", "--tags", "--abbrev=0"]).strip())[2:-1]
|
|
hsh = str(subprocess.check_output(["git", "rev-parse", "--short", "HEAD"]).strip())[2:-1]
|
|
update = str(subprocess.check_output(["git", "log", "-1", "--format=%cd"]).strip())[2:-7]
|
|
except:
|
|
hsh="Unknown"
|
|
update="Unknown"
|
|
tag="Unknown"
|
|
return render_template('status.html', title='STATUS', count=count, max=config['maxInstanceUsers'], file=filen,
|
|
cani=caniregister, hash=hsh, update=update, tag=tag)
|
|
|
|
|
|
@app.route('/error/<errno>')
|
|
def error(errno):
|
|
return render_template('{}.html'.format(str(errno)), config=config)
|
|
|
|
|
|
def getTimeDiff(t):
|
|
diff = datetime.datetime.now() - datetime.datetime(*t[:6])
|
|
|
|
if diff.days == 0:
|
|
if diff.seconds > 3599:
|
|
timeString = "{}h".format(int((diff.seconds / 60) / 60))
|
|
else:
|
|
timeString = "{}m".format(int(diff.seconds / 60))
|
|
else:
|
|
timeString = "{}d".format(diff.days)
|
|
return timeString
|
|
|
|
|
|
def isTwitterUser(username):
|
|
response = requests.get('{instance}{user}/rss'.format(instance=NITTERINSTANCE, user=username))
|
|
if response.status_code == 404:
|
|
return False
|
|
return True
|
|
|
|
|
|
def twitterUserSearch(terms):
|
|
response = urllib.request.urlopen(
|
|
'{instance}search?f=users&q={user}'.format(instance=NITTERINSTANCE, user=urllib.parse.quote(terms))).read()
|
|
html = BeautifulSoup(str(response), "lxml")
|
|
|
|
results = []
|
|
if html.body.find('h2', attrs={'class': 'timeline-none'}):
|
|
return False
|
|
else:
|
|
html = html.body.find_all('div', attrs={'class': 'timeline-item'})
|
|
for item in html:
|
|
user = {
|
|
"fullName": item.find('a', attrs={'class': 'fullname'}).getText().encode('latin_1').decode(
|
|
'unicode_escape').encode('latin_1').decode('utf8'),
|
|
"username": item.find('a', attrs={'class': 'username'}).getText().encode('latin_1').decode(
|
|
'unicode_escape').encode('latin_1').decode('utf8'),
|
|
'avatar': "{i}{s}".format(i=NITTERINSTANCE, s=item.find('img', attrs={'class': 'avatar'})['src'][1:])
|
|
}
|
|
results.append(user)
|
|
return results
|
|
|
|
|
|
def getTwitterUserInfo(username):
|
|
response = urllib.request.urlopen('{instance}{user}'.format(instance=NITTERINSTANCE, user=username)).read()
|
|
# rssFeed = feedparser.parse(response.content)
|
|
|
|
html = BeautifulSoup(str(response), "lxml")
|
|
if html.body.find('div', attrs={'class': 'error-panel'}):
|
|
return False
|
|
else:
|
|
html = html.body.find('div', attrs={'class': 'profile-card'})
|
|
|
|
if html.find('a', attrs={'class': 'profile-card-fullname'}):
|
|
fullName = html.find('a', attrs={'class': 'profile-card-fullname'}).getText().encode('latin1').decode(
|
|
'unicode_escape').encode('latin1').decode('utf8')
|
|
else:
|
|
fullName = None
|
|
|
|
if html.find('div', attrs={'class': 'profile-bio'}):
|
|
profileBio = html.find('div', attrs={'class': 'profile-bio'}).getText().encode('latin1').decode(
|
|
'unicode_escape').encode('latin1').decode('utf8')
|
|
else:
|
|
profileBio = None
|
|
|
|
user = {
|
|
"profileFullName": fullName,
|
|
"profileUsername": html.find('a', attrs={'class': 'profile-card-username'}).string.encode('latin_1').decode(
|
|
'unicode_escape').encode('latin_1').decode('utf8'),
|
|
"profileBio": profileBio,
|
|
"tweets": html.find_all('span', attrs={'class': 'profile-stat-num'})[0].string,
|
|
"following": html.find_all('span', attrs={'class': 'profile-stat-num'})[1].string,
|
|
"followers": numerize.numerize(
|
|
int(html.find_all('span', attrs={'class': 'profile-stat-num'})[2].string.replace(",", ""))),
|
|
"likes": html.find_all('span', attrs={'class': 'profile-stat-num'})[3].string,
|
|
"profilePic": "{instance}{pic}".format(instance=NITTERINSTANCE,
|
|
pic=html.find('a', attrs={'class': 'profile-card-avatar'})['href'][
|
|
1:])
|
|
}
|
|
return user
|
|
|
|
|
|
def getFeed(urls):
|
|
feedPosts = []
|
|
with FuturesSession() as session:
|
|
futures = [session.get('{instance}{user}'.format(instance=NITTERINSTANCE, user=u.username)) for u in urls]
|
|
for future in as_completed(futures):
|
|
res= future.result().content
|
|
html = BeautifulSoup(res, "html.parser")
|
|
userFeed = html.find_all('div', attrs={'class': 'timeline-item'})
|
|
if userFeed != []:
|
|
for post in userFeed[:-1]:
|
|
date_time_str = post.find('span', attrs={'class': 'tweet-date'}).find('a')['title'].replace(",", "")
|
|
time = datetime.datetime.now() - datetime.datetime.strptime(date_time_str, '%d/%m/%Y %H:%M:%S')
|
|
if time.days >= 7:
|
|
continue
|
|
|
|
if post.find('div', attrs={'class': 'pinned'}):
|
|
if post.find('div', attrs={'class': 'pinned'}).find('span', attrs={'icon-pin'}):
|
|
continue
|
|
|
|
newPost = {}
|
|
newPost["op"] = post.find('a', attrs={'class': 'username'}).text
|
|
newPost["twitterName"] = post.find('a', attrs={'class': 'fullname'}).text
|
|
newPost["timeStamp"] = date_time_str
|
|
newPost["date"] = post.find('span', attrs={'class': 'tweet-date'}).find('a').text
|
|
content = post.find('div', attrs={'class': 'tweet-content'})
|
|
newPost["content"] = Markup(str(content).replace("\n", "<br>"))
|
|
|
|
if post.find('div', attrs={'class': 'retweet-header'}):
|
|
newPost["username"] = post.find('div', attrs={'class': 'retweet-header'}).find('div', attrs={
|
|
'class': 'icon-container'}).text
|
|
newPost["isRT"] = True
|
|
else:
|
|
newPost["username"] = newPost["op"]
|
|
newPost["isRT"] = False
|
|
|
|
newPost["profilePic"] = NITTERINSTANCE + \
|
|
post.find('a', attrs={'class': 'tweet-avatar'}).find('img')['src'][1:]
|
|
newPost["url"] = NITTERINSTANCE + post.find('a', attrs={'class': 'tweet-link'})['href'][1:]
|
|
if post.find('div', attrs={'class': 'quote'}):
|
|
newPost["isReply"] = True
|
|
quote = post.find('div', attrs={'class': 'quote'})
|
|
if quote.find('div', attrs={'class': 'quote-text'}):
|
|
newPost["replyingTweetContent"] = Markup(quote.find('div', attrs={'class': 'quote-text'}))
|
|
|
|
if quote.find('a', attrs={'class': 'still-image'}):
|
|
newPost["replyAttachedImg"] = NITTERINSTANCE + \
|
|
quote.find('a', attrs={'class': 'still-image'})['href'][1:]
|
|
|
|
if quote.find('div', attrs={'class': 'unavailable-quote'}):
|
|
newPost["replyingUser"] = "Unavailable"
|
|
else:
|
|
try:
|
|
newPost["replyingUser"] = quote.find('a', attrs={'class': 'username'}).text
|
|
except:
|
|
newPost["replyingUser"] = "Unavailable"
|
|
post.find('div', attrs={'class': 'quote'}).decompose()
|
|
|
|
if post.find('div', attrs={'class': 'attachments'}):
|
|
if not post.find(class_='quote'):
|
|
if post.find('div', attrs={'class': 'attachments'}).find('a',
|
|
attrs={'class': 'still-image'}):
|
|
newPost["attachedImg"] = NITTERINSTANCE + \
|
|
post.find('div', attrs={'class': 'attachments'}).find('a')[
|
|
'href'][1:]
|
|
feedPosts.append(newPost)
|
|
return feedPosts
|
|
|
|
|
|
def getPosts(account):
|
|
feedPosts = []
|
|
|
|
# Gather profile info.
|
|
rssFeed = urllib.request.urlopen('{instance}{user}'.format(instance=NITTERINSTANCE, user=account)).read()
|
|
# Gather feedPosts
|
|
res = rssFeed.decode('utf-8')
|
|
html = BeautifulSoup(res, "html.parser")
|
|
userFeed = html.find_all('div', attrs={'class': 'timeline-item'})
|
|
if userFeed != []:
|
|
for post in userFeed[:-1]:
|
|
date_time_str = post.find('span', attrs={'class': 'tweet-date'}).find('a')['title'].replace(",", "")
|
|
|
|
if post.find('div', attrs={'class': 'pinned'}):
|
|
if post.find('div', attrs={'class': 'pinned'}).find('span', attrs={'icon-pin'}):
|
|
continue
|
|
|
|
newPost = twitterPost()
|
|
newPost.op = post.find('a', attrs={'class': 'username'}).text
|
|
newPost.twitterName = post.find('a', attrs={'class': 'fullname'}).text
|
|
newPost.timeStamp = datetime.datetime.strptime(date_time_str, '%d/%m/%Y %H:%M:%S')
|
|
newPost.date = post.find('span', attrs={'class': 'tweet-date'}).find('a').text
|
|
newPost.content = Markup(post.find('div', attrs={'class': 'tweet-content'}))
|
|
|
|
if post.find('div', attrs={'class': 'retweet-header'}):
|
|
newPost.username = post.find('div', attrs={'class': 'retweet-header'}).find('div', attrs={
|
|
'class': 'icon-container'}).text
|
|
newPost.isRT = True
|
|
else:
|
|
newPost.username = newPost.op
|
|
newPost.isRT = False
|
|
|
|
newPost.profilePic = NITTERINSTANCE + post.find('a', attrs={'class': 'tweet-avatar'}).find('img')['src'][1:]
|
|
newPost.url = NITTERINSTANCE + post.find('a', attrs={'class': 'tweet-link'})['href'][1:]
|
|
if post.find('div', attrs={'class': 'quote'}):
|
|
newPost.isReply = True
|
|
quote = post.find('div', attrs={'class': 'quote'})
|
|
if quote.find('div', attrs={'class': 'quote-text'}):
|
|
newPost.replyingTweetContent = Markup(quote.find('div', attrs={'class': 'quote-text'}))
|
|
|
|
if quote.find('a', attrs={'class': 'still-image'}):
|
|
newPost.replyAttachedImg = NITTERINSTANCE + quote.find('a', attrs={'class': 'still-image'})['href'][
|
|
1:]
|
|
|
|
try:
|
|
newPost.replyingUser = quote.find('a', attrs={'class': 'username'}).text
|
|
except:
|
|
newPost.replyingUser = "Unavailable"
|
|
post.find('div', attrs={'class': 'quote'}).decompose()
|
|
|
|
if post.find('div', attrs={'class': 'attachments'}):
|
|
if not post.find(class_='quote'):
|
|
if post.find('div', attrs={'class': 'attachments'}).find('a', attrs={'class': 'still-image'}):
|
|
newPost.attachedImg = NITTERINSTANCE + \
|
|
post.find('div', attrs={'class': 'attachments'}).find('a')['href'][1:]
|
|
feedPosts.append(newPost)
|
|
return feedPosts
|
|
|
|
|
|
def getYoutubePosts(ids):
|
|
videos = []
|
|
with FuturesSession() as session:
|
|
futures = [session.get('https://www.youtube.com/feeds/videos.xml?channel_id={id}'.format(id=id.channelId)) for
|
|
id in ids]
|
|
for future in as_completed(futures):
|
|
resp = future.result()
|
|
rssFeed = feedparser.parse(resp.content)
|
|
for vid in rssFeed.entries:
|
|
try:
|
|
# Try to get time diff
|
|
time = datetime.datetime.now() - datetime.datetime(*vid.published_parsed[:6])
|
|
except:
|
|
# If youtube rss does not have parsed time, generate it. Else set time to 0.
|
|
try:
|
|
time = datetime.datetime.now() - datetime.datetime(
|
|
datetime.datetime.strptime(vid.published, '%y-%m-%dT%H:%M:%S+00:00'))
|
|
except:
|
|
time = datetime.datetime.now() - datetime.datetime.now()
|
|
|
|
if time.days >= 6:
|
|
continue
|
|
|
|
video = ytPost()
|
|
try:
|
|
video.date = vid.published_parsed
|
|
except:
|
|
try:
|
|
video.date = datetime.datetime.strptime(vid.published, '%y-%m-%dT%H:%M:%S+00:00').timetuple()
|
|
except:
|
|
video.date = datetime.datetime.utcnow().timetuple()
|
|
try:
|
|
video.timeStamp = getTimeDiff(vid.published_parsed)
|
|
except:
|
|
if time != 0:
|
|
video.timeStamp = "{} days".format(str(time.days))
|
|
else:
|
|
video.timeStamp = "Unknown"
|
|
|
|
video.channelName = vid.author_detail.name
|
|
video.channelId = vid.yt_channelid
|
|
video.channelUrl = vid.author_detail.href
|
|
video.id = vid.yt_videoid
|
|
video.videoTitle = vid.title
|
|
if config['isInstance']:
|
|
hostName = urllib.parse.urlparse(vid.media_thumbnail[0]['url']).netloc
|
|
video.videoThumb = vid.media_thumbnail[0]['url'].replace("https://{}".format(hostName), "").replace(
|
|
"hqdefault", "mqdefault") + "?host=" + hostName
|
|
else:
|
|
video.videoThumb = vid.media_thumbnail[0]['url'].replace('/', '~')
|
|
video.views = vid.media_statistics['views']
|
|
video.description = vid.summary_detail.value
|
|
video.description = re.sub(r'^https?:\/\/.*[\r\n]*', '', video.description[0:120] + "...",
|
|
flags=re.MULTILINE)
|
|
videos.append(video)
|
|
return videos |