This repository has been archived on 2022-06-28. You can view files and clone it, but cannot push or open issues or pull requests.
Yotter/app/routes.py

852 lines
34 KiB
Python
Raw Normal View History

2020-07-31 15:58:52 +05:30
from flask import render_template, flash, redirect, url_for, request, send_from_directory, Markup
2020-07-27 18:30:01 +05:30
from app.forms import LoginForm, RegistrationForm, EmptyForm, SearchForm, ChannelForm
from app.models import User, twitterPost, ytPost, Post, youtubeFollow, twitterFollow
2020-07-13 03:13:36 +05:30
from flask_login import login_user, logout_user, current_user, login_required
2020-08-16 18:50:05 +05:30
from flask import Flask, Response, stream_with_context
2020-07-14 21:15:29 +05:30
from requests_futures.sessions import FuturesSession
2020-08-30 02:05:29 +05:30
from werkzeug.datastructures import Headers
2020-07-14 21:15:29 +05:30
from concurrent.futures import as_completed
from werkzeug.utils import secure_filename
2020-08-17 14:32:18 +05:30
from youtube_search import YoutubeSearch
2020-07-13 03:13:36 +05:30
from werkzeug.urls import url_parse
from youtube_dl import YoutubeDL
2020-08-27 02:43:54 +05:30
from numerize import numerize
2020-08-16 18:50:05 +05:30
from bs4 import BeautifulSoup
from xml.dom import minidom
2020-07-13 03:13:36 +05:30
from app import app, db
2020-08-30 02:05:29 +05:30
from re import findall
2020-07-13 03:13:36 +05:30
import random, string
2020-07-31 15:58:52 +05:30
import time, datetime
2020-07-13 03:13:36 +05:30
import feedparser
2020-07-13 20:06:45 +05:30
import requests
2020-08-30 13:35:16 +05:30
import bleach
2020-08-27 01:43:51 +05:30
import urllib
2020-09-20 13:04:53 +05:30
import math
2020-07-27 18:30:01 +05:30
import json
import re
2020-09-11 14:04:24 +05:30
#########################################
from youtube_data import videos as ytvids
from youtube_data import search as yts
#########################################
2020-08-30 04:45:48 +05:30
##########################
#### Config variables ####
##########################
2020-09-04 11:51:35 +05:30
config = json.load(open('yotter-config.json'))
##########################
#### Config variables ####
##########################
NITTERINSTANCE = config['nitterInstance'] # Must be https://.../
2020-08-30 04:45:48 +05:30
YOUTUBERSS = "https://www.youtube.com/feeds/videos.xml?channel_id="
2020-07-13 03:13:36 +05:30
##########################
#### Global variables ####
##########################
ALLOWED_EXTENSIONS = {'json', 'db'}
2020-07-31 15:58:52 +05:30
#########################
#### Twitter Logic ######
#########################
@app.before_request
def before_request():
if current_user.is_authenticated:
current_user.set_last_seen()
db.session.commit()
2020-07-13 03:13:36 +05:30
@app.route('/')
@app.route('/index')
@login_required
def index():
2020-09-08 16:38:05 +05:30
return render_template('home.html', config=config)
2020-08-26 13:11:34 +05:30
@app.route('/twitter')
@login_required
def twitter():
2020-07-31 15:58:52 +05:30
start_time = time.time()
followingList = current_user.twitter_following_list()
followCount = len(followingList)
2020-07-13 03:13:36 +05:30
posts = []
avatarPath = "img/avatars/1.png"
2020-07-13 20:06:45 +05:30
form = EmptyForm()
posts.extend(getFeed(followingList))
posts.sort(key=lambda x: x.timeStamp, reverse=True)
2020-07-14 21:15:29 +05:30
if not posts:
profilePic = avatarPath
else:
profilePic = posts[0].userProfilePic
2020-07-31 15:58:52 +05:30
print("--- {} seconds fetching twitter feed---".format(time.time() - start_time))
2020-09-08 16:38:05 +05:30
return render_template('twitter.html', title='Yotter | Twitter', posts=posts, avatar=avatarPath, profilePic = profilePic, followedCount=followCount, form=form, config=config)
2020-07-13 03:13:36 +05:30
2020-07-13 20:06:45 +05:30
@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.")
2020-08-27 01:43:51 +05:30
return redirect(request.referrer)
2020-07-13 20:06:45 +05:30
@app.route('/saved')
@login_required
def saved():
savedPosts = current_user.saved_posts().all()
2020-09-08 16:38:05 +05:30
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'))
2020-07-13 03:13:36 +05:30
@app.route('/follow/<username>', methods=['POST'])
@login_required
def follow(username):
form = EmptyForm()
if form.validate_on_submit():
2020-08-24 19:01:56 +05:30
if followTwitterAccount(username):
flash("{} followed!".format(username))
return redirect(request.referrer)
2020-08-24 19:01:56 +05:30
def followTwitterAccount(username):
if isTwitterUser(username):
2020-08-24 19:01:56 +05:30
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
2020-07-13 03:13:36 +05:30
else:
flash("Something went wrong... try again")
return False
2020-07-13 03:13:36 +05:30
@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)
2020-07-13 03:13:36 +05:30
def twUnfollow(username):
try:
user = twitterFollow.query.filter_by(username=username).first()
db.session.delete(user)
2020-07-13 05:40:51 +05:30
db.session.commit()
except:
flash("There was an error unfollowing the user. Try again.")
return redirect(request.referrer)
2020-07-13 05:40:51 +05:30
@app.route('/following')
@login_required
def following():
form = EmptyForm()
followCount = len(current_user.twitter_following_list())
accounts = current_user.twitter_following_list()
2020-09-08 16:38:05 +05:30
return render_template('following.html', accounts = accounts, count = followCount, form = form, config=config)
2020-07-13 05:40:51 +05:30
@app.route('/search', methods=['GET', 'POST'])
@login_required
def search():
form = SearchForm()
if form.validate_on_submit():
user = form.username.data
2020-08-27 13:37:48 +05:30
results = twitterUserSearch(user)
2020-07-14 22:30:50 +05:30
if results:
2020-09-08 16:38:05 +05:30
return render_template('search.html', form = form, results = results, config=config)
2020-07-13 05:40:51 +05:30
else:
2020-07-14 22:30:50 +05:30
flash("User {} not found...".format(user))
2020-08-27 13:37:48 +05:30
return redirect(request.referrer)
2020-07-13 05:40:51 +05:30
else:
2020-09-08 16:38:05 +05:30
return render_template('search.html', form = form, config=config)
2020-07-13 05:40:51 +05:30
2020-08-27 15:37:59 +05:30
@app.route('/u/<username>')
2020-09-08 13:37:06 +05:30
@app.route('/<username>')
2020-07-13 03:13:36 +05:30
@login_required
2020-08-27 15:37:59 +05:30
def u(username):
2020-09-26 23:36:24 +05:30
if len(username.split("/")) > 1:
2020-08-27 01:43:51 +05:30
form = EmptyForm()
avatarPath = "img/avatars/{}.png".format(str(random.randint(1,12)))
user = getTwitterUserInfo(username)
if not user:
flash("This user is not on Twitter.")
2020-08-27 01:43:51 +05:30
return redirect(request.referrer)
2020-07-13 05:40:51 +05:30
2020-07-13 03:13:36 +05:30
posts = []
posts.extend(getPosts(username))
2020-07-14 21:15:29 +05:30
if not posts:
2020-08-27 01:43:51 +05:30
user['profilePic'] = avatarPath
2020-09-08 16:38:05 +05:30
return render_template('user.html', posts=posts, user=user, form=form, config=config)
2020-07-13 03:13:36 +05:30
2020-07-31 15:58:52 +05:30
#########################
#### Youtube Logic ######
#########################
@app.route('/youtube', methods=['GET', 'POST'])
2020-07-31 15:58:52 +05:30
@login_required
def youtube():
followCount = len(current_user.youtube_following_list())
2020-07-31 15:58:52 +05:30
start_time = time.time()
ids = current_user.youtube_following_list()
2020-08-16 18:50:05 +05:30
videos = getYoutubePosts(ids)
2020-07-31 15:58:52 +05:30
if videos:
videos.sort(key=lambda x: x.date, reverse=True)
2020-08-23 20:57:12 +05:30
print("--- {} seconds fetching youtube feed---".format(time.time() - start_time))
2020-09-08 16:38:05 +05:30
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)
2020-09-08 16:38:05 +05:30
return render_template('ytfollowing.html', form=form, channelList=channelList, channelCount=channelCount, config=config)
2020-07-31 15:58:52 +05:30
@app.route('/ytsearch', methods=['GET', 'POST'])
@login_required
def ytsearch():
form = ChannelForm()
button_form = EmptyForm()
if form.validate_on_submit():
2020-09-11 14:04:24 +05:30
searchTerms = form.channelId.data
page = 1
autocorrect = 1
sort = 0
filters = {"time":0, "type":0, "duration":0}
results = yts.search_by_terms(searchTerms, page, autocorrect, sort, filters)
return render_template('ytsearch.html', form=form, btform=button_form, results=results, restricted=config['restrictPublicUsage'], config=config)
2020-08-17 14:32:18 +05:30
2020-07-31 15:58:52 +05:30
else:
2020-09-11 14:10:02 +05:30
return render_template('ytsearch.html', form=form, results=False)
2020-07-31 15:58:52 +05:30
@app.route('/ytfollow/<channelId>', methods=['POST'])
@login_required
def ytfollow(channelId):
form = EmptyForm()
if form.validate_on_submit():
r = followYoutubeChannel(channelId)
return redirect(request.referrer)
def followYoutubeChannel(channelId):
channelData = YoutubeSearch.channelInfo(channelId, False)
try:
2020-08-24 19:01:56 +05:30
if not current_user.is_following_yt(channelId):
follow = youtubeFollow()
follow.channelId = channelId
follow.channelName = channelData[0]['name']
follow.followers.append(current_user)
db.session.add(follow)
db.session.commit()
flash("{} followed!".format(channelData[0]['name']))
return True
else:
return False
2020-09-20 16:23:25 +05:30
except Exception as e:
print(e)
2020-08-24 19:01:56 +05:30
flash("Youtube: Couldn't follow {}. Already followed?".format(channelData[0]['name']))
return False
2020-07-31 15:58:52 +05:30
@app.route('/ytunfollow/<channelId>', methods=['POST'])
@login_required
def ytunfollow(channelId):
form = EmptyForm()
if form.validate_on_submit():
unfollowYoutubeChannel(channelId)
return redirect(request.referrer)
def unfollowYoutubeChannel(channelId):
2020-07-31 15:58:52 +05:30
try:
channel = youtubeFollow.query.filter_by(channelId=channelId).first()
name = channel.channelName
2020-07-31 15:58:52 +05:30
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))
2020-07-31 15:58:52 +05:30
except:
flash("There was an error unfollowing the user. Try again.")
@app.route('/channel/<id>', methods=['GET'])
2020-08-27 15:37:59 +05:30
@app.route('/user/<id>', methods=['GET'])
@app.route('/c/<id>', methods=['GET'])
@login_required
def channel(id):
2020-08-23 21:06:00 +05:30
form = ChannelForm()
button_form = EmptyForm()
data = requests.get('https://www.youtube.com/feeds/videos.xml?channel_id={id}'.format(id=id))
data = feedparser.parse(data.content)
2020-08-23 14:33:49 +05:30
channelData = YoutubeSearch.channelInfo(id)
2020-09-26 17:37:37 +05:30
for video in channelData[1]:
print(video)
if config['nginxVideoStream']:
hostName = urllib.parse.urlparse(video['videoThumb']).netloc
2020-09-26 17:49:03 +05:30
video['videoThumb'] = video['videoThumb'].replace("https://{}".format(hostName), "").replace("hqdefault", "mqdefault")+"&host="+hostName
2020-09-26 17:37:37 +05:30
else:
video['videoThumb'] = video['videoThumb'].replace('/', '~')
if config['nginxVideoStream']:
hostName = urllib.parse.urlparse(channelData[0]['avatar']).netloc
channelData[0]['avatar'] = channelData[0]['avatar'].replace("https://{}".format(hostName), "")+"?host="+hostName
else:
channelData[0]['avatar'] = channelData[0]['avatar'].replace('/', '~')
2020-09-26 17:49:03 +05:30
2020-09-08 16:38:05 +05:30
return render_template('channel.html', form=form, btform=button_form, channel=channelData[0], videos=channelData[1], restricted=config['restrictPublicUsage'], config=config)
2020-08-17 11:56:56 +05:30
@app.route('/watch', methods=['GET'])
2020-07-31 15:58:52 +05:30
@login_required
def watch():
id = request.args.get('v', None)
2020-09-10 05:54:51 +05:30
info = ytvids.get_video_info(id)
hostName = urllib.parse.urlparse(info['video']['url']).netloc
2020-09-20 15:29:26 +05:30
audioHostName = urllib.parse.urlparse(info['video']['audio']['url']).netloc
# Use nginx
try:
2020-09-20 15:33:05 +05:30
url = info['video']['url'].replace("https://{}".format(hostName), "")+"&host="+hostName
except:
url = "#"
2020-09-20 15:29:26 +05:30
try:
2020-09-20 15:33:05 +05:30
audioUrl = info['video']['audio']['url'].replace("https://{}".format(audioHostName), "")+"&host="+audioHostName
2020-09-20 15:29:26 +05:30
print(audioUrl)
except:
audioUrl = False
2020-09-20 15:49:42 +05:30
print(info['video']['thumbnail'])
video={
2020-09-10 05:54:51 +05:30
'title':info['video']['title'],
'description':Markup(markupString(info['video']['description'])),
'viewCount':info['video']['views'],
'author':info['video']['author'],
'authorUrl':"/channel/{}".format(info['owner']['id']),
'channelId': info['owner']['id'],
'id':id,
2020-09-10 05:54:51 +05:30
'averageRating': str((float(info['video']['rating'])/5)*100),
'nginxUrl': url,
2020-09-10 05:54:51 +05:30
'videoUrl': info['video']['url'],
'videoHostName': hostName,
2020-09-10 05:54:51 +05:30
'isLive': info['video']['isLive'],
'isUpcoming': info['video']['isUpcoming'],
2020-09-20 15:29:26 +05:30
'thumbnail': info['video']['thumbnail'],
'nginxAudioUrl': audioUrl
2020-07-31 15:58:52 +05:30
}
2020-09-08 16:38:05 +05:30
return render_template("video.html", video=video, title='{}'.format(video['title']), config=config)
2020-07-31 15:58:52 +05:30
2020-08-30 13:35:16 +05:30
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)
2020-08-30 13:35:16 +05:30
2020-08-30 16:10:20 +05:30
## PROXY videos through Yotter server to the client.
@app.route('/stream/<url>', methods=['GET', 'POST'])
@login_required
def stream(url):
2020-08-30 02:05:29 +05:30
#This function proxies the video stream from GoogleVideo to the client.
url = url.replace('YotterSlash', '/')
headers = Headers()
if(url):
2020-09-08 13:43:27 +05:30
s = requests.Session()
s.verify = True
req = s.get(url, stream = True)
headers.add('Range', request.headers['Range'])
2020-08-30 02:05:29 +05:30
headers.add('Accept-Ranges','bytes')
headers.add('Content-Length', str(int(req.headers['Content-Length'])+1))
2020-09-08 13:40:59 +05:30
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)
2020-08-30 02:05:29 +05:30
#enable browser file caching with etags
response.cache_control.public = True
response.cache_control.max_age = int(60000)
return response
else:
2020-08-17 11:56:56 +05:30
flash("Something went wrong loading the video... Try again.")
return redirect(url_for('youtube'))
2020-08-30 02:05:29 +05:30
def download_file(streamable):
with streamable as stream:
stream.raise_for_status()
for chunk in stream.iter_content(chunk_size=8192):
yield chunk
2020-07-31 15:58:52 +05:30
#########################
#### 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'))
2020-09-20 13:04:53 +05:30
if user.username == config['admin_user']:
user.set_admin_user()
db.session.commit()
2020-07-31 15:58:52 +05:30
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)
2020-09-08 16:38:05 +05:30
return render_template('login.html', title='Sign In', form=form, config=config)
2020-07-31 15:58:52 +05:30
2020-09-04 11:12:01 +05:30
#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")
2020-07-31 15:58:52 +05:30
@app.route('/logout')
def logout():
logout_user()
return redirect(url_for('index'))
@app.route('/settings')
@login_required
def settings():
2020-09-08 16:38:05 +05:30
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:
2020-09-08 16:44:55 +05:30
t = datetime.datetime.utcnow() - u.last_seen
s = t.total_seconds()
m = s/60
2020-09-26 23:36:24 +05:30
if m < 25:
active = active+1
2020-09-05 16:55:45 +05:30
instanceInfo = {
"totalUsers":db.session.query(User).count(),
"active":active,
2020-09-05 16:55:45 +05:30
}
2020-09-20 13:04:53 +05:30
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)
2020-07-31 15:58:52 +05:30
@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():
2020-08-24 19:01:56 +05:30
twitterFollowing = current_user.twitter_following_list()
2020-07-31 15:58:52 +05:30
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')
2020-08-31 16:58:57 +05:30
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')
2020-08-31 16:58:57 +05:30
return redirect(request.referrer)
if file and allowed_file(file.filename) or 'subscription_manager' in file.filename:
option = request.form['import_format']
if option == 'yotter':
importYotterSubscriptions(file)
elif option == 'newpipe':
importNewPipeSubscriptions(file)
elif option == 'youtube':
importYoutubeSubscriptions(file)
elif option == 'freetube':
importFreeTubeSubscriptions(file)
2020-08-31 16:58:57 +05:30
return redirect(request.referrer)
2020-08-24 19:01:56 +05:30
return redirect(request.referrer)
2020-09-06 13:57:14 +05:30
@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 importYotterSubscriptions(file):
2020-08-24 19:01:56 +05:30
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'])
def importNewPipeSubscriptions(file):
filename = secure_filename(file.filename)
data = json.load(file)
for acc in data['subscriptions']:
r = followYoutubeChannel(re.search('(UC[a-zA-Z0-9_-]{22})|(?<=user\/)[a-zA-Z0-9_-]+', acc['url']).group())
def importYoutubeSubscriptions(file):
filename = secure_filename(file.filename)
itemlist = minidom.parse(file).getElementsByTagName('outline')
for item in itemlist[1:]:
r = followYoutubeChannel(re.search('UC[a-zA-Z0-9_-]{22}', item.attributes['xmlUrl'].value).group())
def importFreeTubeSubscriptions(file):
filename = secure_filename(file.filename)
data = re.findall('UC[a-zA-Z0-9_-]{22}', file.read().decode('utf-8'))
for acc in data:
r = followYoutubeChannel(acc)
def allowed_file(filename):
return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
2020-07-31 15:58:52 +05:30
@app.route('/register', methods=['GET', 'POST'])
def register():
2020-09-04 21:51:57 +05:30
form = RegistrationForm()
2020-07-31 15:58:52 +05:30
if current_user.is_authenticated:
return redirect(url_for('index'))
2020-09-04 20:37:44 +05:30
REGISTRATIONS = True
2020-09-04 21:17:41 +05:30
try:
count = db.session.query(User).count()
if count >= config['maxInstanceUsers'] or config['maxInstanceUsers'] == 0:
REGISTRATIONS = False
except:
REGISTRATIONS = True
2020-09-04 21:51:57 +05:30
2020-07-31 15:58:52 +05:30
if form.validate_on_submit():
2020-08-27 13:45:20 +05:30
if User.query.filter_by(username=form.username.data).first():
flash("This username is taken! Try with another.")
return redirect(request.referrer)
2020-08-27 14:26:00 +05:30
2020-08-27 13:45:20 +05:30
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'))
2020-09-04 21:51:57 +05:30
2020-09-08 16:38:05 +05:30
return render_template('register.html', title='Register', registrations=REGISTRATIONS, form=form, config=config)
2020-09-26 18:39:26 +05:30
@app.route('/registrations_status/icon')
def registrations_status_icon():
count = db.session.query(User).count()
if count >= config['maxInstanceUsers'] or config['maxInstanceUsers'] == 0:
2020-09-26 18:39:26 +05:30
return redirect(url_for('static',filename='img/close.png'))
else:
2020-09-26 18:39:26 +05:30
return redirect(url_for('static',filename='img/open.png'))
@app.route('/registrations_status/text')
def registrations_status_text():
count = db.session.query(User).count()
return "{c}/{t}".format(c=count, t=config['maxInstanceUsers'])
2020-07-31 15:58:52 +05:30
@app.route('/error/<errno>')
def error(errno):
2020-09-08 16:38:05 +05:30
return render_template('{}.html'.format(str(errno)), config=config)
2020-07-31 15:58:52 +05:30
2020-07-13 03:13:36 +05:30
def getTimeDiff(t):
2020-08-27 02:43:54 +05:30
diff = datetime.datetime.now() - datetime.datetime(*t[:6])
2020-07-13 03:13:36 +05:30
if diff.days == 0:
2020-07-13 20:06:45 +05:30
if diff.seconds > 3599:
timeString = "{}h".format(int((diff.seconds/60)/60))
2020-07-13 03:13:36 +05:30
else:
2020-07-13 20:06:45 +05:30
timeString = "{}m".format(int(diff.seconds/60))
2020-07-13 03:13:36 +05:30
else:
timeString = "{}d".format(diff.days)
2020-07-13 05:40:51 +05:30
return timeString
def isTwitterUser(username):
2020-08-30 04:45:48 +05:30
response = requests.get('{instance}{user}/rss'.format(instance=NITTERINSTANCE, user=username))
2020-08-27 01:43:51 +05:30
if response.status_code == 404:
2020-07-13 05:40:51 +05:30
return False
return True
2020-08-27 13:37:48 +05:30
def twitterUserSearch(terms):
2020-08-28 03:26:14 +05:30
2020-08-30 04:45:48 +05:30
response = urllib.request.urlopen('{instance}search?f=users&q={user}'.format(instance=NITTERINSTANCE, user=urllib.parse.quote(terms))).read()
2020-08-27 13:37:48 +05:30
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'),
2020-08-30 04:45:48 +05:30
'avatar': "{i}{s}".format(i=NITTERINSTANCE, s=item.find('img', attrs={'class':'avatar'})['src'][1:])
2020-08-27 13:37:48 +05:30
}
results.append(user)
return results
2020-08-27 01:43:51 +05:30
def getTwitterUserInfo(username):
2020-08-30 04:45:48 +05:30
response = urllib.request.urlopen('{instance}{user}'.format(instance=NITTERINSTANCE, user=username)).read()
2020-08-27 01:43:51 +05:30
#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'})
2020-08-27 13:37:48 +05:30
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
2020-08-27 01:43:51 +05:30
user = {
2020-08-27 13:37:48 +05:30
"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,
2020-08-27 01:43:51 +05:30
"tweets":html.find_all('span', attrs={'class':'profile-stat-num'})[0].string,
"following":html.find_all('span', attrs={'class':'profile-stat-num'})[1].string,
2020-08-27 02:43:54 +05:30
"followers":numerize.numerize(int(html.find_all('span', attrs={'class':'profile-stat-num'})[2].string.replace(",",""))),
2020-08-27 01:43:51 +05:30
"likes":html.find_all('span', attrs={'class':'profile-stat-num'})[3].string,
2020-08-30 04:45:48 +05:30
"profilePic":"{instance}{pic}".format(instance=NITTERINSTANCE, pic=html.find('a', attrs={'class':'profile-card-avatar'})['href'][1:])
2020-08-27 01:43:51 +05:30
}
return user
2020-07-14 21:15:29 +05:30
def getFeed(urls):
feedPosts = []
with FuturesSession() as session:
2020-08-30 04:45:48 +05:30
futures = [session.get('{instance}{user}'.format(instance=NITTERINSTANCE, user=u.username)) for u in urls]
2020-07-14 21:15:29 +05:30
for future in as_completed(futures):
2020-08-28 02:55:08 +05:30
res = future.result().content.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(",","")
time = datetime.datetime.now() - datetime.datetime.strptime(date_time_str, '%d/%m/%Y %H:%M:%S')
2020-09-04 11:12:01 +05:30
if time.days >=7:
2020-08-27 01:43:51 +05:30
continue
2020-08-28 02:55:08 +05:30
2020-08-28 03:26:14 +05:30
if post.find('div', attrs={'class':'pinned'}):
if post.find('div', attrs={'class':'pinned'}).find('span', attrs={'icon-pin'}):
continue
2020-07-14 21:15:29 +05:30
newPost = twitterPost()
2020-08-28 02:55:08 +05:30
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
2020-09-08 16:48:08 +05:30
newPost.content = Markup(post.find('div', attrs={'class':'tweet-content'}))
2020-07-14 21:15:29 +05:30
2020-08-28 02:55:08 +05:30
if post.find('div', attrs={'class':'retweet-header'}):
newPost.username = post.find('div', attrs={'class':'retweet-header'}).find('div', attrs={'class':'icon-container'}).text
2020-07-14 21:15:29 +05:30
newPost.isRT = True
else:
2020-08-28 02:55:08 +05:30
newPost.username = newPost.op
2020-07-14 21:15:29 +05:30
newPost.isRT = False
2020-08-28 02:55:08 +05:30
2020-08-30 04:45:48 +05:30
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:]
2020-08-28 02:55:08 +05:30
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'}):
2020-08-30 04:45:48 +05:30
newPost.replyAttachedImg = NITTERINSTANCE+quote.find('a', attrs={'class':'still-image'})['href'][1:]
2020-08-28 02:55:08 +05:30
2020-09-09 14:56:56 +05:30
if quote.find('div', attrs={'class':'unavailable-quote'}):
newPost.replyingUser="Unavailable"
else:
newPost.replyingUser=quote.find('a', attrs={'class':'username'}).text
2020-08-28 02:55:08 +05:30
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'}):
2020-08-30 04:45:48 +05:30
newPost.attachedImg = NITTERINSTANCE + post.find('div', attrs={'class':'attachments'}).find('a')['href'][1:]
2020-07-14 21:15:29 +05:30
feedPosts.append(newPost)
return feedPosts
def getPosts(account):
2020-09-02 00:33:51 +05:30
feedPosts = []
#Gather profile info.
2020-09-02 00:33:51 +05:30
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:]
newPost.replyingUser=quote.find('a', attrs={'class':'username'}).text
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
2020-07-27 18:30:01 +05:30
2020-08-16 18:50:05 +05:30
def getYoutubePosts(ids):
2020-07-27 18:30:01 +05:30
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]
2020-07-27 18:30:01 +05:30
for future in as_completed(futures):
resp = future.result()
rssFeed=feedparser.parse(resp.content)
for vid in rssFeed.entries:
2020-09-09 17:32:45 +05:30
try:
2020-09-10 03:55:00 +05:30
# Try to get time diff
2020-09-09 17:32:45 +05:30
time = datetime.datetime.now() - datetime.datetime(*vid.published_parsed[:6])
except:
2020-09-10 05:54:51 +05:30
# 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()
2020-09-10 05:54:51 +05:30
if time.days >=6:
continue
2020-08-30 13:35:16 +05:30
2020-08-16 18:50:05 +05:30
video = ytPost()
2020-09-09 17:32:45 +05:30
try:
video.date = vid.published_parsed
except:
2020-09-10 05:54:51 +05:30
try:
video.date = datetime.datetime.strptime(vid.published, '%y-%m-%dT%H:%M:%S+00:00').timetuple()
except:
video.date = datetime.datetime.utcnow().timetuple()
2020-09-09 17:32:45 +05:30
try:
video.timeStamp = getTimeDiff(vid.published_parsed)
except:
2020-09-10 05:54:51 +05:30
if time != 0:
video.timeStamp = "{} days".format(str(time.days))
else:
video.timeStamp = "Unknown"
2020-07-27 18:30:01 +05:30
video.channelName = vid.author_detail.name
2020-08-23 20:57:12 +05:30
video.channelId = vid.yt_channelid
2020-07-27 18:30:01 +05:30
video.channelUrl = vid.author_detail.href
2020-08-16 18:50:05 +05:30
video.id = vid.yt_videoid
2020-07-27 18:30:01 +05:30
video.videoTitle = vid.title
2020-09-15 00:09:10 +05:30
if config['nginxVideoStream']:
hostName = urllib.parse.urlparse(vid.media_thumbnail[0]['url']).netloc
2020-09-20 15:49:42 +05:30
video.videoThumb = vid.media_thumbnail[0]['url'].replace("https://{}".format(hostName), "").replace("hqdefault", "mqdefault")+"?host="+hostName
2020-09-15 00:09:10 +05:30
else:
video.videoThumb = vid.media_thumbnail[0]['url'].replace('/', '~')
video.views = vid.media_statistics['views']
2020-08-16 18:50:05 +05:30
video.description = vid.summary_detail.value
2020-07-27 18:30:01 +05:30
video.description = re.sub(r'^https?:\/\/.*[\r\n]*', '', video.description[0:120]+"...", flags=re.MULTILINE)
videos.append(video)
2020-09-04 19:29:16 +05:30
return videos