Skip to content

Commit

Permalink
feat(models): add add and remove api for infini package model
Browse files Browse the repository at this point in the history
  • Loading branch information
fu050409 committed Mar 8, 2024
1 parent 31272d8 commit 46a82e2
Showing 1 changed file with 11 additions and 1 deletion.
12 changes: 11 additions & 1 deletion src/ipm/models/ipk.py
Original file line number Diff line number Diff line change
Expand Up @@ -161,6 +161,16 @@ def unrequire(self, name: str) -> None:
raise ProjectError(f"规则包 [bold green]{name}[/] 不在规则包依赖中.")
self._data["requirements"].remove(name) # type: ignore

def add(self, name: str, version: str) -> None:
denpendencies = self.dependencies
denpendencies.update({name: version})
self._data["dependencies"] = denpendencies

def remove(self, name: str) -> None:
if name not in self._data.get("dependencies", {}):
raise ProjectError(f"规则包 [bold green]{name}[/] 不在规则包依赖中.")
self._data["dependencies"].remove(name) # type: ignore

@property
def plain_dict(self) -> TOMLDocument:
return self._data
Expand All @@ -187,7 +197,7 @@ def license(self) -> str:

@property
def dependencies(self) -> Dict[str, Any]:
return self._data["dependencies"] # type: ignore
return self._data.get("dependencies", {}) # type: ignore

@property
def requirements(self) -> Requirements:
Expand Down

0 comments on commit 46a82e2

Please sign in to comment.