Changed database scripts

This commit is contained in:
StNicolay
2022-10-14 15:38:03 +03:00
parent e09c56759e
commit 3ebe4240bf
8 changed files with 116 additions and 113 deletions

View File

@ -1,20 +1,24 @@
import os
import mariadb
from dotenv import load_dotenv
from sqlalchemy.future import Engine
from . import bot, cryptography, database
__all__ = ["bot", "cryptography", "database"]
engine: Engine
def main() -> None:
global engine
load_dotenv("./.env")
con = mariadb.connect(
os.getenv("DB_HOST"),
os.getenv("DB_USER"),
os.getenv("DB_PASS"),
os.getenv("DB_NAME"),
)
database.prepare(con)
engine = database.prepare.get_engine(
host=os.getenv("DB_HOST"),
user=os.getenv("DB_USER"),
passwd=os.getenv("DB_PASS"),
db=os.getenv("DB_NAME"),
) # type: ignore
database.prepare.prepare(engine)
bot_ = bot.create_bot(os.getenv("TG_TOKEN"), con) # type: ignore
bot_.infinity_polling()