Skip to content

Commit 51e9b70

Browse files
committed
add DBDeployment module
1 parent a0f0faf commit 51e9b70

1 file changed

Lines changed: 26 additions & 0 deletions

File tree

forloop_common_structures/database_models.py

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -684,6 +684,32 @@ def cast_script_types_to_db(scripts_df: pd.DataFrame) -> pd.DataFrame:
684684
return scripts_df
685685

686686

687+
class DBDeployment(dh.AbstractModel):
688+
pipeline_uid: str
689+
port: int
690+
host: str = "0.0.0.0"
691+
module_app: str = "user_api_sample:app"
692+
subprocess_id: int = None
693+
status: str = None
694+
created_at: datetime = Field(default_factory=datetime.utcnow)
695+
updated_at: datetime = None
696+
697+
698+
def cast_deployment_types_to_app(deployments_df: pd.DataFrame) -> pd.DataFrame:
699+
"""Cast DB datatypes to in-app python datatypes."""
700+
deployments_df = gdtm.cast_types_to_app(deployments_df)
701+
deployments_df[["created_at", "updated_at"]] = deployments_df[["created_at", "updated_at"]].astype(object).replace(pd.NaT, None)
702+
return deployments_df
703+
704+
705+
def cast_deployment_types_to_db(deployments_df: pd.DataFrame) -> pd.DataFrame:
706+
"""Cast in-app python datatypes to DB datatypes."""
707+
deployments_df = gdtm.cast_types_to_db(deployments_df)
708+
deployments_df[["created_at", "updated_at"]] = deployments_df[["created_at", "updated_at"]].astype(object).replace(pd.NaT, None)
709+
deployments_df = deployments_df.map(escape_if_string)
710+
return deployments_df
711+
712+
687713
class DBDatabase(dh.AbstractModel):
688714
database_name: str
689715
server: str

0 commit comments

Comments
 (0)