Code refactor
This commit is contained in:
@@ -1,6 +1,14 @@
|
|||||||
from psycopg.rows import dict_row
|
from psycopg.rows import dict_row
|
||||||
from .connection import g_conn_dict as conn
|
from .connection import g_conn_dict as conn
|
||||||
|
|
||||||
|
def _get_parents(name: str, id: int) -> list[int]:
|
||||||
|
ids = [id]
|
||||||
|
with conn[name].cursor(row_factory=dict_row) as cur:
|
||||||
|
while ids[-1] != 0:
|
||||||
|
cur.execute(f"select parent from operation where id = {ids[-1]}")
|
||||||
|
ids.append(int(cur.fetchone()['parent']))
|
||||||
|
return ids
|
||||||
|
|
||||||
def _get_current_operation(name: str) -> int:
|
def _get_current_operation(name: str) -> int:
|
||||||
with conn[name].cursor(row_factory=dict_row) as cur:
|
with conn[name].cursor(row_factory=dict_row) as cur:
|
||||||
cur.execute(f"select id from current_operation")
|
cur.execute(f"select id from current_operation")
|
||||||
@@ -90,15 +98,6 @@ def take_snapshot(name: str, tag: str) -> None:
|
|||||||
parent = _get_current_operation(name)
|
parent = _get_current_operation(name)
|
||||||
cur.execute(f"insert into snapshot_operation (id, tag) values ({curr}, '{tag}')")
|
cur.execute(f"insert into snapshot_operation (id, tag) values ({curr}, '{tag}')")
|
||||||
|
|
||||||
|
|
||||||
def _get_parents(name: str, id: int) -> list[int]:
|
|
||||||
ids = [id]
|
|
||||||
with conn[name].cursor(row_factory=dict_row) as cur:
|
|
||||||
while ids[-1] != 0:
|
|
||||||
cur.execute(f"select parent from operation where id = {ids[-1]}")
|
|
||||||
ids.append(int(cur.fetchone()['parent']))
|
|
||||||
return ids
|
|
||||||
|
|
||||||
def pick_snapshot(name: str, tag: str) -> None:
|
def pick_snapshot(name: str, tag: str) -> None:
|
||||||
if tag == None or tag == '':
|
if tag == None or tag == '':
|
||||||
print('Non empty tag is expected!')
|
print('Non empty tag is expected!')
|
||||||
|
|||||||
Reference in New Issue
Block a user