diff options
| -rw-r--r-- | src/ipm/models/lock.py | 77 |
1 files changed, 61 insertions, 16 deletions
diff --git a/src/ipm/models/lock.py b/src/ipm/models/lock.py index 92a7a2d..4422738 100644 --- a/src/ipm/models/lock.py +++ b/src/ipm/models/lock.py @@ -78,11 +78,56 @@ class PackageLock(IpmLock): def __init__(self) -> None: super().__init__() + def add_index( + self, index_uri: str, host: str, uuid: str, dump: bool = False + ) -> str: + for index in self.indexes: + if "index" not in index.keys(): + raise SyntaxError("异常的锁文件!") + if index["index"] == index_uri: + self.storages.remove(index) + break + + self.indexes.append( + { + "index": index_uri, + "host": host, + "uuid": uuid, + } + ) + return self.dump() if dump else "" + + def remove_index(self, uuid: str, dump: bool = False) -> str: + uuid = uuid.strip() + for index in self.indexes: + if "name" not in index.keys(): + raise SyntaxError("异常的锁文件!") + if index["name"] == uuid: + self.packages.remove(index) + break + return self.dump() if dump else "" + + def get_index(self, index_uri: str) -> dict | None: + index_uri = index_uri.strip() + for index in self.indexes: + if index["index"] == index_uri: + return index + return None + + def has_index(self, index_uri: str) -> bool: + index_uri = index_uri.strip() + for index in self.indexes: + if index["index"] == index_uri: + return True + return False + def add_package(self, ipk: "ipk.InfiniProject", dump: bool = False) -> str: for package in self.packages: if "name" not in package.keys(): raise SyntaxError("异常的锁文件!") - if package["name"] == ipk.name and package["version"] == ipk.version: + if ( + package["name"] == ipk.name and package["version"] == ipk.version + ): # TODO 同名包处理 self.storages.remove(package) break @@ -97,6 +142,20 @@ class PackageLock(IpmLock): ) return self.dump() if dump else "" + def get_package(self, name: str) -> dict | None: + name = name.strip() + for package in self.packages: + if package["name"] == name: + return package + return None + + def has_package(self, name: str) -> bool: + name = name.strip() + for package in self.packages: + if package["name"] == name: + return True + return False + def remove_package(self, name: str, dump: bool = False) -> str: name = name.strip() for package in self.packages: @@ -135,20 +194,6 @@ class PackageLock(IpmLock): break return self.dump() if dump else "" - def get_package(self, name: str) -> dict | None: - name = name.strip() - for package in self.packages: - if package["name"] == name: - return package - return None - - def has_package(self, name: str) -> bool: - name = name.strip() - for package in self.packages: - if package["name"] == name: - return True - return False - def get_storage(self, name: str) -> dict | None: name = name.strip() for storage in self.storages: @@ -156,7 +201,7 @@ class PackageLock(IpmLock): return storage return None - def has_storage(self, name: str) -> dict | None: + def has_storage(self, name: str) -> bool: name = name.strip() for storage in self.storages: if storage["name"] == name: |