matu
/
Heroku
mirror of https://github.com/coddrago/Heroku
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix data root in web

pull/1/head
hikari.ftg 2022-04-01 18:24:33 +00:00
parent 54122dbb23
commit 9fa316cd94
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
hikka/web/root.py
View File

@ -39,6 +39,12 @@ from .. import utils, main
from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton
DATA_DIR = (
os.path.normpath(os.path.join(utils.get_base_dir(), ".."))
if "OKTETO" not in os.environ
else "/data"
)
class Web: class Web:
def __init__(self, **kwargs): def __init__(self, **kwargs):
@ -101,9 +107,7 @@ class Web:
): ):
return web.Response(status=400) return web.Response(status=400)
with open( with open(
os.path.join( os.path.join(self.data_root or DATA_DIR, "api_token.txt"),
self.data_root or os.path.dirname(utils.get_base_dir()), "api_token.txt"
),
"w", "w",
) as f: ) as f:
f.write(api_id + "\n" + api_hash) f.write(api_id + "\n" + api_hash)
@ -209,7 +213,8 @@ class Web:
markup = InlineKeyboardMarkup() markup = InlineKeyboardMarkup()
markup.add( markup.add(
InlineKeyboardButton( InlineKeyboardButton(
"🔓 Authorize user", callback_data=f"authorize_web_{token}" "🔓 Authorize user",
callback_data=f"authorize_web_{token}",
) )
) )