Changed database scripts
This commit is contained in:
@ -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()
|
||||
|
Reference in New Issue
Block a user