Skip to content

Commit

Permalink
feat(models): move defaults out of base models
Browse files Browse the repository at this point in the history
  • Loading branch information
RobertRosca committed Sep 24, 2024
1 parent 5cceb83 commit 5a41f83
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 7 deletions.
4 changes: 2 additions & 2 deletions src/zulip_write_only_proxy/models/base.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
import datetime
from abc import ABC

from pydantic import BaseModel, Field
from pydantic import BaseModel


class Base(ABC, BaseModel):
created_at: datetime.datetime = Field(default_factory=datetime.datetime.now)
created_at: datetime.datetime

@property
def _key(self):
Expand Down
15 changes: 11 additions & 4 deletions src/zulip_write_only_proxy/models/client.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
import datetime
import secrets
from typing import IO, TYPE_CHECKING, Any

Expand All @@ -21,10 +22,16 @@
class ScopedClientCreate(BaseModel):
proposal_no: int
stream: str | None = None
bot_name: str | None = None
bot_email: str | None = None
bot_key: str | None = None
bot_site: str = "https://mylog.connect.xfel.eu/"
bot_id: int | None = None

token: SecretStr = Field(
default_factory=lambda: SecretStr(secrets.token_urlsafe()), init_var=False
)

created_at: datetime.datetime = Field(
default_factory=datetime.datetime.now, init_var=False
)


class ScopedClient(Base):
Expand All @@ -33,7 +40,7 @@ class ScopedClient(Base):
bot_name: str
bot_id: int
bot_site: HttpUrl
token: SecretStr = Field(default_factory=lambda: SecretStr(secrets.token_urlsafe()))
token: SecretStr

# created_at - from base
created_by: str
Expand Down
2 changes: 1 addition & 1 deletion src/zulip_write_only_proxy/models/zulip.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ class BotConfig(Base):
id: int
proposal_no: int

# created_at - from base
created_at: datetime | None = None

@property
def _key(self):
Expand Down

0 comments on commit 5a41f83

Please sign in to comment.