94 lines
3 KiB
Python
94 lines
3 KiB
Python
from flask import Flask, request, redirect, session, render_template
|
|
import twitter_credentials, tweepy, database
|
|
|
|
app = Flask(__name__)
|
|
app.secret_key = "*i9uld6u@t!kxl9%o+byxqf14&a&&@y@q=l$!lg4m%b-a*^o(a"
|
|
|
|
def is_twkeys(session):
|
|
try:
|
|
session["access_token"]
|
|
session["access_secret_token"]
|
|
except:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
def twapi_login(session):
|
|
auth = tweepy.OAuthHandler(twitter_credentials.consumer_key, twitter_credentials.consumer_secret_key)
|
|
auth.set_access_token(session["access_token"], session["access_secret_token"])
|
|
return tweepy.API(auth)
|
|
|
|
def is_twlogin(session):
|
|
return is_twkeys(session) and twapi_login(session).verify_credentials()
|
|
|
|
def is_trlogin(session):
|
|
return is_twkeys(session) and database.trtoken(twapi_login(session).me().id)
|
|
|
|
@app.route("/twlogin")
|
|
def twlogin():
|
|
auth = tweepy.OAuthHandler(twitter_credentials.consumer_key, twitter_credentials.consumer_secret_key, "https://cyberplanificateur.flifloo.fr/twlogin")
|
|
if request.args.get("oauth_token") and request.args.get("oauth_verifier"):
|
|
auth.request_token = {"oauth_token" : request.args.get("oauth_token"), "oauth_token_secret" : request.args.get("oauth_verifier")}
|
|
try:
|
|
auth.get_access_token(request.args.get("oauth_verifier"))
|
|
except:
|
|
return "Error ! Failed to get access token"
|
|
else:
|
|
session["access_token"] = auth.access_token
|
|
session["access_secret_token"] = auth.access_token_secret
|
|
elif not is_twlogin(session):
|
|
return redirect(auth.get_authorization_url())
|
|
return redirect("/")
|
|
|
|
@app.route("/twlogout")
|
|
def twlogout():
|
|
if is_twkeys(session):
|
|
session.pop("access_token", None)
|
|
session.pop("access_secret_token", None)
|
|
return redirect("/")
|
|
|
|
@app.route("/trlogin")
|
|
def trlogin():
|
|
if not is_twlogin(session):
|
|
return redirect("/")
|
|
if request.args.get("token"):
|
|
database.tradd(twapi_login(session).me().id, request.args.get("token"))
|
|
return redirect("/settings")
|
|
|
|
@app.route("/trlogout")
|
|
def trlogout():
|
|
if not is_twlogin(session):
|
|
return redirect("/")
|
|
if is_trlogin(session):
|
|
database.trrm(twapi_login(session).me().id)
|
|
return redirect("/settings")
|
|
|
|
@app.route("/")
|
|
def home():
|
|
return render_template("index.html", login = is_twlogin(session))
|
|
|
|
@app.route("/settings", methods = ['POST', 'GET'])
|
|
def settings():
|
|
if not is_twlogin(session):
|
|
return redirect("/")
|
|
api = twapi_login(session)
|
|
if "token" in request.form:
|
|
database.tradd(api.me().id, request.form["token"])
|
|
return redirect("/settings")
|
|
return render_template("settings.html", login = True, trlogin = is_trlogin(session))
|
|
|
|
@app.route("/twpost")
|
|
def twpost():
|
|
if is_twlogin(session):
|
|
api = twapi_login(session)
|
|
api.update_status("bloup")
|
|
return "Send !"
|
|
else:
|
|
return "Not login !"
|
|
|
|
@app.route("/test")
|
|
def test():
|
|
return render_template("elements.html")
|
|
|
|
if __name__ == "__main__":
|
|
app.run(debug=True, port=5000, host="0.0.0.0")
|