mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-23 19:44:20 +00:00
48 lines
No EOL
1.4 KiB
Python
48 lines
No EOL
1.4 KiB
Python
from flask import Flask
|
|
from flask import session as flask_session
|
|
from flask_openid import OpenID
|
|
from db import session, Royal, Steam
|
|
import requests
|
|
import re
|
|
|
|
# Init the config reader
|
|
import configparser
|
|
config = configparser.ConfigParser()
|
|
config.read("config.ini")
|
|
|
|
# Init Flask and Flask_OpenID
|
|
app = Flask(__name__)
|
|
app.secret_key = config["Steam"]["secret_key"]
|
|
oid = OpenID(app)
|
|
|
|
@app.route("/login/<int:royal_id>")
|
|
@oid.loginhandler
|
|
def page_steam_login(royal_id):
|
|
flask_session["royal_id"] = royal_id
|
|
return oid.try_login("http://steamcommunity.com/openid")
|
|
|
|
|
|
@oid.after_login
|
|
def page_after_login(response):
|
|
steam_id = re.search("https?://steamcommunity\.com/openid/id/(.+)", response.identity_url).group(1)
|
|
db_steam = session.query(Steam).filter(Steam.steam_id == steam_id).first()
|
|
if db_steam is None:
|
|
db_steam = Steam(royal_id=flask_session["royal_id"],
|
|
steam_id=steam_id)
|
|
session.add(db_steam)
|
|
db_steam.update()
|
|
session.commit()
|
|
return "Account Steam collegato con successo!"
|
|
else:
|
|
return "Il tuo account Steam è già collegato."
|
|
|
|
|
|
@app.route("/update/<int:royal_id>")
|
|
def page_steam_update(royal_id):
|
|
db_steam = session.query(Steam).filter(Steam.royal_id == royal_id).first()
|
|
db_steam.update()
|
|
return "Dati account aggiornati."
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host="0.0.0.0", port="1234") |