ryg/royalnet
ryg/royalnet
1
Fork 0
mirror of https://github.com/RYGhub/royalnet.git synced 2024-11-24 03:54:20 +00:00
Code Activity

tglogin progress

Browse source
This commit is contained in:
Steffo 2019-06-08 01:33:34 +02:00
parent 7fd241e1fd
commit d6ac988a9c
6 changed files with 57 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
royalnet/royalgamesweb.py
View file

@ -1,13 +1,13 @@
import os import os
from .web import create_app from .web import create_app
from .web.blueprints import home, wikiview from .web.blueprints import home, wikiview, tglogin
class TestConfig: class TestConfig:
DB_PATH = os.environ["DB_PATH"] DB_PATH = os.environ["DB_PATH"]
app = create_app(TestConfig, [home, wikiview]) app = create_app(TestConfig, [home, wikiview, tglogin])
if __name__ == "__main__": if __name__ == "__main__":

3
royalnet/web/blueprints/__init__.py
View file

@ -2,5 +2,6 @@ from .helloworld import bp as helloworld
from .testing import bp as testing from .testing import bp as testing
from .home import bp as home from .home import bp as home
from .wikiview import bp as wikiview from .wikiview import bp as wikiview
from .tglogin import bp as tglogin
__all__ = ["helloworld", "testing", "home", "wikiview"] __all__ = ["helloworld", "testing", "home", "wikiview", "tglogin"]

11
royalnet/web/blueprints/helloworld.py
View file

@ -1,11 +0,0 @@
import flask as f
from .. import Royalprint
bp = Royalprint("helloworld", __name__, url_prefix="/helloworld")
@bp.route("/")
def helloworld():
return "Hello world!"

13
royalnet/web/blueprints/testing.py
View file

@ -1,13 +0,0 @@
import flask as f
from .. import Royalprint
from ...database.tables import Royal
bp = Royalprint("testing", __name__, url_prefix="/testing", required_tables={Royal})
@bp.route("/listroyals")
def listroyals():
from ..alchemyhandler import alchemy, alchemy_session
royals = alchemy_session.query(alchemy.Royal).all()
return f'<body><script type="text/plain" style="display: block;">{repr(royals)}</script></body>'

35
royalnet/web/blueprints/tglogin/__init__.py Normal file
View file

@ -0,0 +1,35 @@
import flask as f
import hashlib
import hmac
from ... import Royalprint
from ....database.tables import Royal, Telegram
bp = Royalprint("tglogin", __name__, url_prefix="/login/telegram", required_tables={Royal, Telegram},
template_folder="templates")
@bp.route("/")
def tglogin_index():
return f.render_template("tglogin_index.html")
@bp.route("/done")
def tglogin_done():
from ...alchemyhandler import alchemy, alchemy_session
data_check_string = ""
for field in f.request.args:
if field == "hash":
continue
data_check_string += f"{field}={f.request.args['field']}\n"
data_check_string.rstrip("\n")
secret_key = hashlib.sha256(f.current_app.config["TG_AK"])
hex_data = hmac.new(key=secret_key, msg=data_check_string, digestmod="sha256").hexdigest()
if hex_data != f.request.args["hash"]:
return "Invalid authentication", 403
tg_user = alchemy_session.query(alchemy.Telegram).filter(alchemy.Telegram.tg_id == f.request.args["id"]).one_or_none()
if tg_user is None:
return "No such telegram", 404
royal_user = tg_user.royal
f.session["royal_id"] = royal_user.id
return f.redirect(f.url_for("/"))

18
royalnet/web/blueprints/tglogin/templates/tglogin_index.html Normal file
View file

@ -0,0 +1,18 @@
{% extends "base.html" %}
{% block title %}
Login with Telegram
{% endblock %}
{% block content %}
<div class="doublebox">
<div class="top">
<span class="left">
Telegram login
</span>
</div>
<div class="bot">
<script async src="https://telegram.org/js/telegram-widget.js?6" data-telegram-login="royalgamesbot" data-size="large" data-auth-url="{{ url_for("tglogin.tglogin_done") }}" data-request-access="write"></script>
</div>
</div>
{% endblock %}