You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
29 lines
926 B
29 lines
926 B
from ispoof.objects import Pokemon, Raid
|
|
from ispoof.lists.playerhistory import PlayerHistory
|
|
from sqlalchemy.exc import ArgumentError
|
|
from sqlalchemy import create_engine
|
|
from ispoof.utils import get_home_folder
|
|
import logging
|
|
from sqlalchemy.engine import Engine
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class Database:
|
|
def __init__(self):
|
|
self.data = get_home_folder() / "data"
|
|
self.data.mkdir(parents=True, exist_ok=True)
|
|
self.engine = create_engine(f"sqlite:///{(self.data / 'data.db').absolute()}")
|
|
|
|
def initialize_database(self):
|
|
logger.info("Creating databases.")
|
|
try:
|
|
Pokemon.metadata.create_all(self.engine)
|
|
Raid.metadata.create_all(self.engine)
|
|
PlayerHistory.metadata.create_all(self.engine)
|
|
except ArgumentError:
|
|
logger.info("Databases existed.")
|
|
|
|
def get_engine(self) -> Engine:
|
|
return self.engine
|