Change env name.

pull/1079/head
Andrey 2024-06-03 22:17:57 +03:00
rodzic 565e24231e
commit 7de295bf0d
2 zmienionych plików z 13 dodań i 9 usunięć

Wyświetl plik

@ -14,13 +14,17 @@ logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
try: try:
MOONSTREAM_DB_URI = os.environ.get("MOONSTREAM_DB_URI") MOONSTREAM_DB_V3_INDEXES_URI = os.environ.get("MOONSTREAM_DB_V3_INDEXES_URI")
if MOONSTREAM_DB_URI is None: if MOONSTREAM_DB_V3_INDEXES_URI is None:
raise Warning("MOONSTREAM_DB_URI environment variable must be set") raise Warning("MOONSTREAM_DB_V3_INDEXES_URI environment variable must be set")
MOONSTREAM_DB_URI_READ_ONLY = os.environ.get("MOONSTREAM_DB_URI_READ_ONLY") MOONSTREAM_DB_V3_INDEXES_URI_READ_ONLY = os.environ.get(
if MOONSTREAM_DB_URI_READ_ONLY is None: "MOONSTREAM_DB_V3_INDEXES_URI_READ_ONLY"
raise Warning("MOONSTREAM_DB_URI_READ_ONLY environment variable must be set") )
if MOONSTREAM_DB_V3_INDEXES_URI_READ_ONLY is None:
raise Warning(
"MOONSTREAM_DB_V3_INDEXES_URI_READ_ONLY environment variable must be set"
)
except ValueError as e: except ValueError as e:
raise ValueError(e) raise ValueError(e)
except Warning: except Warning:
@ -87,7 +91,7 @@ class DBEngine:
class MoonstreamDBEngine(DBEngine): class MoonstreamDBEngine(DBEngine):
def __init__(self, schema: Optional[str] = None) -> None: def __init__(self, schema: Optional[str] = None) -> None:
super().__init__(url=MOONSTREAM_DB_URI, schema=schema) super().__init__(url=MOONSTREAM_DB_V3_INDEXES_URI, schema=schema)
self._session_local = sessionmaker(bind=self.engine) self._session_local = sessionmaker(bind=self.engine)
@ -118,7 +122,7 @@ class MoonstreamDBEngine(DBEngine):
class MoonstreamDBEngineRO(DBEngine): class MoonstreamDBEngineRO(DBEngine):
def __init__(self, schema: Optional[str] = None) -> None: def __init__(self, schema: Optional[str] = None) -> None:
super().__init__(url=MOONSTREAM_DB_URI_READ_ONLY, schema=schema) super().__init__(url=MOONSTREAM_DB_V3_INDEXES_URI_READ_ONLY, schema=schema)
@property @property
def engine(self): def engine(self):

Wyświetl plik

@ -1 +1 @@
0.0.7 0.0.8