Skip to content

Commit

Permalink
Revert db change
Browse files Browse the repository at this point in the history
  • Loading branch information
Wh1isper committed Nov 24, 2023
1 parent 9c1bf63 commit a75db55
Showing 1 changed file with 22 additions and 22 deletions.
44 changes: 22 additions & 22 deletions duetector/db.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,7 @@
from __future__ import annotations

from contextlib import contextmanager
from datetime import datetime
from threading import Lock
from typing import Any, Generator
from typing import Any, Dict, Generator, List, Optional

import sqlalchemy # type: ignore
from sqlalchemy.orm import ( # type: ignore
Expand All @@ -30,16 +28,16 @@ class TrackingMixin:
autoincrement=True,
)

pid: Mapped[int | None]
uid: Mapped[int | None]
gid: Mapped[int | None]
dt: Mapped[datetime | None]
pid: Mapped[Optional[int]]
uid: Mapped[Optional[int]]
gid: Mapped[Optional[int]]
dt: Mapped[Optional[datetime]]

comm: Mapped[str | None]
cwd: Mapped[str | None]
fname: Mapped[str | None]
comm: Mapped[Optional[str]]
cwd: Mapped[Optional[str]]
fname: Mapped[Optional[str]]

extended: Mapped[dict[str, Any]] = mapped_column(type_=JSON, default={})
extended: Mapped[Dict[str, Any]] = mapped_column(type_=JSON, default={})

def __repr__(self):
return f"<Tracking [{self.pid} {self.comm}] {self.dt}>"
Expand All @@ -66,7 +64,7 @@ def to_analyzer_tracking(self) -> AT:
def inspect_fields(
cls,
value_as_type: bool = False,
) -> dict[str, Any]:
) -> Dict[str, Any]:
raise NotImplementedError


Expand Down Expand Up @@ -121,11 +119,11 @@ def __repr__(self):

return f"<[SessionManager {safe_url}]{self.table_prefix}*>"

def __init__(self, config: dict[str, Any] | None = None, *args, **kwargs):
def __init__(self, config: Optional[Dict[str, Any]] = None, *args, **kwargs):
super().__init__(config, *args, **kwargs)
self._engine: sqlalchemy.engine.Engine | None = None
self._sessionmaker: sessionmaker | None = None
self._tracking_models: dict[str, type] = {}
self._engine: Optional[sqlalchemy.engine.Engine] = None
self._sessionmaker: Optional[sessionmaker] = None
self._tracking_models: Dict[str, type] = {}
self.mutex = Lock()

@property
Expand All @@ -141,7 +139,7 @@ def table_prefix(self):
return self.config.table_prefix

@property
def engine_config(self) -> dict[str, Any]:
def engine_config(self) -> Dict[str, Any]:
"""
Config for sqlalchemy.create_engine
"""
Expand Down Expand Up @@ -255,7 +253,7 @@ def to_analyzer_tracking(self) -> AT:
def inspect_fields(
cls,
value_as_type: bool = False,
) -> dict[str, Any]:
) -> Dict[str, Any]:
return {
c.name: c.type.python_type if value_as_type else c.type.python_type.__name__
for c in cls.__table__.columns
Expand All @@ -269,10 +267,12 @@ def inspect_fields(
raise
return self._tracking_models[tracer]

def get_all_models(self) -> dict[str, type]:
def get_all_models(self) -> Dict[str, type]:
return self._tracking_models

def inspect_all_tables(self, tracer: str | None = None, collector_id: str | None = None) -> str:
def inspect_all_tables(
self, tracer: Optional[str] = None, collector_id: Optional[str] = None
) -> str:
def _filter(t):
if tracer and self.table_name_to_tracer(t) != tracer:
return False
Expand All @@ -286,10 +286,10 @@ def _filter(t):
if t.startswith(self.table_prefix) and _filter(t)
]

def inspect_all_tracers(self) -> list[str]:
def inspect_all_tracers(self) -> List[str]:
return list(set(self.table_name_to_tracer(t) for t in self.inspect_all_tables()))

def inspect_all_collector_ids(self) -> list[str]:
def inspect_all_collector_ids(self) -> List[str]:
return list(set(self.table_name_to_collector_id(t) for t in self.inspect_all_tables()))

def _init_tracking_model(self, tracking_model: type) -> type:
Expand Down

0 comments on commit a75db55

Please sign in to comment.