class TaskGroupV2(SQLModel, table=True):
id: int | None = Field(default=None, primary_key=True)
task_list: list[TaskV2] = Relationship(
sa_relationship_kwargs=dict(
lazy="selectin", cascade="all, delete-orphan"
),
)
user_id: int = Field(foreign_key="user_oauth.id")
user_group_id: int | None = Field(
foreign_key="usergroup.id", default=None, ondelete="SET NULL"
)
origin: str
pkg_name: str
version: str | None = None
python_version: str | None = None
pixi_version: str | None = None
path: str | None = None
archive_path: str | None = None
pip_extras: str | None = None
pinned_package_versions: dict[str, str] = Field(
sa_column=Column(
JSONB,
server_default="{}",
default={},
nullable=True,
),
)
env_info: str | None = None
venv_path: str | None = None
venv_size_in_kB: int | None = None
venv_file_number: int | None = None
active: bool = True
timestamp_created: datetime = Field(
default_factory=get_timestamp,
sa_column=Column(DateTime(timezone=True), nullable=False),
)
timestamp_last_used: datetime = Field(
default_factory=get_timestamp,
sa_column=Column(
DateTime(timezone=True),
nullable=False,
server_default=(
datetime(2024, 11, 20, tzinfo=timezone.utc).isoformat()
),
),
)
@property
def pip_install_string(self) -> str:
"""
Prepare string to be used in `python -m pip install`.
"""
if self.origin == "pixi":
raise ValueError(
f"Cannot call 'pip_install_string' if {self.origin=}."
)
extras = f"[{self.pip_extras}]" if self.pip_extras is not None else ""
if self.archive_path is not None:
return f"{self.archive_path}{extras}"
else:
if self.version is None:
raise ValueError(
"Cannot run `pip_install_string` with "
f"{self.pkg_name=}, {self.archive_path=}, {self.version=}."
)
return f"{self.pkg_name}{extras}=={self.version}"
@property
def pinned_package_versions_string(self) -> str:
"""
Prepare string to be used in `python -m pip install`.
"""
if self.origin == "pixi":
raise ValueError(
"Cannot call 'pinned_package_versions_string' if "
f"{self.origin=}."
)
if self.pinned_package_versions is None:
return ""
output = " ".join(
[
f"{key}=={value}"
for key, value in self.pinned_package_versions.items()
]
)
return output