-
Notifications
You must be signed in to change notification settings - Fork 0
database
Provides database connection and SQL queries throught ORM.
class DatabaseType(StrEnum)Enum to name databases
class DatabaseConnection()Class to hold database engine.
def init_db(db_path, db_type) -> NoneInit an DatabaseConnection, if there is no tables, creates then
def insert_credentials(user_id, password)Insert an Credentials model entrie in Client Database.
def insert_hidro(collection: List[HidroBase], has_id=False) -> NoneInsert a list of Hidro ORM Model into Hidro Database
def insert_jobs(jobs: List[HidroJob]) -> NoneInsert a list of Jobs into Client Database.
def update_jobs(jobs: List[HidroJob], job_config: JobConfig) -> NoneUpdates a list of Jobs.
def count_client(model: ClientBase)Counts a given Model in Client Database.
def get_credentials() -> CredentialsGets the first registered Credential on Client Database and returns it.
def add_token(client_id, token, expires)Add an Token to Client Database
def get_token_model() -> TokenReturns the first found Token on Client Database
def update_token(RegistroID, new_token, new_expires)Updates an Token on Client Database
def get_station_jobs(status) -> StationJobsReturns all Stations Jobs on Client Database
def get_jobs_yield(job_config, status)Returns all Series Jobs on Client Database, yield then in batches
def count_job(job_config: JobConfig, status=None)Counts jobs registered in Client Database.
def count_hidro(model: HidroBase)Counts a given Model in Hidro Database.
def get_states() -> StateReturns registered States in Hidro Database.
def get_rain_period()Returns Stations with Rain Periods in Hidro Database.
def get_discharge_period()Returns Stations with Discharge Periods in Hidro Database.
def get_sediments_period()Returns Stations with Sediments Periods in Hidro Database.
def get_water_period()Returns Stations with Water Periods in Hidro Database.
def get_stage_period()Returns Stations with Stage Periods in Hidro Database.