Support cascade deletion

This commit is contained in:
WQY\qiong
2023-02-04 13:45:14 +08:00
parent 633a616d39
commit 804e1c7796
38 changed files with 1272 additions and 707 deletions

View File

@@ -1,4 +1,4 @@
from .operation import *
from .database import *
from .s0_base import *
from .s24_coordinates import *
@@ -47,7 +47,7 @@ class Reservoir(object):
return { 'type': self.type, 'id': self.id }
def set_reservoir_cache(name: str, cs: ChangeSet) -> DbChangeSet:
def set_reservoir_cmd(name: str, cs: ChangeSet) -> DbChangeSet:
old = Reservoir(get_reservoir(name, cs.operations[0]['id']))
raw_new = get_reservoir(name, cs.operations[0]['id'])
@@ -71,10 +71,10 @@ def set_reservoir_cache(name: str, cs: ChangeSet) -> DbChangeSet:
def set_reservoir(name: str, cs: ChangeSet) -> ChangeSet:
return execute_command(name, set_reservoir_cache(name, cs))
return execute_command(name, set_reservoir_cmd(name, cs))
def add_reservoir_cache(name: str, cs: ChangeSet) -> DbChangeSet:
def add_reservoir_cmd(name: str, cs: ChangeSet) -> DbChangeSet:
new = Reservoir(cs.operations[0])
redo_sql = f"insert into _node (id, type) values ({new.f_id}, {new.f_type});"
@@ -92,10 +92,10 @@ def add_reservoir_cache(name: str, cs: ChangeSet) -> DbChangeSet:
def add_reservoir(name: str, cs: ChangeSet) -> ChangeSet:
return execute_command(name, add_reservoir_cache(name, cs))
return execute_command(name, add_reservoir_cmd(name, cs))
def delete_reservoir_cache(name: str, cs: ChangeSet) -> DbChangeSet:
def delete_reservoir_cmd(name: str, cs: ChangeSet) -> DbChangeSet:
old = Reservoir(get_reservoir(name, cs.operations[0]['id']))
redo_sql = f"delete from coordinates where node = {old.f_id};"
@@ -113,7 +113,7 @@ def delete_reservoir_cache(name: str, cs: ChangeSet) -> DbChangeSet:
def delete_reservoir(name: str, cs: ChangeSet) -> ChangeSet:
return execute_command(name, delete_reservoir_cache(name, cs))
return execute_command(name, delete_reservoir_cmd(name, cs))
class InpReservoir:
@@ -151,3 +151,13 @@ def inp_out_reservoir(name: str) -> list[str]:
desc = ';'
lines.append(f'{id} {head} {pattern} {desc}')
return lines
def unset_reservoir_by_pattern(name: str, pattern: str) -> ChangeSet:
cs = ChangeSet()
rows = read_all(name, f"select id from reservoirs where pattern = '{pattern}'")
for row in rows:
cs.append(g_update_prefix | {'type': 'reservoir', 'id': row['id'], 'pattern': None})
return cs