yacloud/shadowmire
1
0
Fork 0
mirror of https://github.com/taoky/shadowmire.git synced 2025-07-08 09:12:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Put serial -1 handling at LocalVersionKV

Browse Source
This commit is contained in:
taoky 2024-08-04 16:51:23 +08:00
parent d92d2079bd
commit 0f97cb7dbd
1 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
shadowmire.py
View File

@ -99,20 +99,26 @@ class LocalVersionKV:
if commit: if commit:
self.conn.commit() self.conn.commit()
def keys(self) -> list[str]: def keys(self, skip_invalid: bool = True) -> list[str]:
cur = self.conn.cursor() cur = self.conn.cursor()
res = cur.execute("SELECT key FROM local") if skip_invalid:
res = cur.execute("SELECT key FROM local WHERE value != -1")
else:
res = cur.execute("SELECT key FROM local")
rows = res.fetchall() rows = res.fetchall()
return [row[0] for row in rows] return [row[0] for row in rows]
def dump(self) -> dict[str, int]: def dump(self, skip_invalid: bool = True) -> dict[str, int]:
cur = self.conn.cursor() cur = self.conn.cursor()
res = cur.execute("SELECT key, value FROM local") if skip_invalid:
res = cur.execute("SELECT key, value FROM local WHERE value != -1")
else:
res = cur.execute("SELECT key, value FROM local")
rows = res.fetchall() rows = res.fetchall()
return {row[0]: row[1] for row in rows} return {row[0]: row[1] for row in rows}
def dump_json(self) -> None: def dump_json(self, skip_invalid: bool = True) -> None:
res = self.dump() res = self.dump(skip_invalid)
with overwrite(self.jsonpath) as f: with overwrite(self.jsonpath) as f:
json.dump(res, f) json.dump(res, f)
@ -396,8 +402,6 @@ class SyncBase:
local_serial = local[i] local_serial = local[i]
remote_serial = remote[i] remote_serial = remote[i]
if local_serial != remote_serial: if local_serial != remote_serial:
if local_serial == -1:
logger.info("skip %s, as it's marked as not exist at upstream", i)
to_update.append(i) to_update.append(i)
output = Plan(remove=to_remove, update=to_update) output = Plan(remove=to_remove, update=to_update)
return output return output
@ -538,7 +542,7 @@ class SyncPyPI(SyncBase):
meta = self.pypi.get_package_metadata(package_name) meta = self.pypi.get_package_metadata(package_name)
logger.debug("%s meta: %s", package_name, meta) logger.debug("%s meta: %s", package_name, meta)
except PackageNotFoundError: except PackageNotFoundError:
logger.warning("%s missing from upstream, skip.", package_name) logger.warning("%s missing from upstream, skip and ignore in the future.", package_name)
if not write_db: if not write_db:
return -1 return -1
self.local_db.set(package_name, -1) self.local_db.set(package_name, -1)