apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-04-29 10:56:24 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

add (backend) google calendar connection flow

Browse source
This commit is contained in:
CREDO23 2025-08-02 00:05:55 +02:00
parent d98dfd40b5
commit 9abaf4fd2a
3 changed files with 176 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

29
surfsense_backend/app/db.py
View file

@ -46,6 +46,7 @@ class DocumentType(str, Enum):
JIRA_CONNECTOR = "JIRA_CONNECTOR"
CONFLUENCE_CONNECTOR = "CONFLUENCE_CONNECTOR"
CLICKUP_CONNECTOR = "CLICKUP_CONNECTOR"
GOOGLE_CALENDAR_CONNECTOR = "GOOGLE_CALENDAR_CONNECTOR"
class SearchSourceConnectorType(str, Enum):
@ -60,6 +61,7 @@ class SearchSourceConnectorType(str, Enum):
JIRA_CONNECTOR = "JIRA_CONNECTOR"
CONFLUENCE_CONNECTOR = "CONFLUENCE_CONNECTOR"
CLICKUP_CONNECTOR = "CLICKUP_CONNECTOR"
GOOGLE_CALENDAR_CONNECTOR = "GOOGLE_CALENDAR_CONNECTOR"
class ChatType(str, Enum):
@ -244,6 +246,21 @@ class SearchSourceConnector(BaseModel, TimestampMixin):
user = relationship("User", back_populates="search_source_connectors")
class GoogleCalendarAccount(BaseModel):
__tablename__ = "google_calendar_accounts"
user_id = Column(
UUID(as_uuid=True),
ForeignKey("user.id", ondelete="CASCADE"),
nullable=False,
unique=True,
)
access_token = Column(String, nullable=False)
refresh_token = Column(String, nullable=False)
user = relationship("User", back_populates="calendar_account")
class LLMConfig(BaseModel, TimestampMixin):
__tablename__ = "llm_configs"
@ -297,6 +314,12 @@ if config.AUTH_TYPE == "GOOGLE":
search_source_connectors = relationship(
"SearchSourceConnector", back_populates="user"
)
calendar_account = relationship(
"GoogleCalendarAccount",
back_populates="user",
uselist=False,
cascade="all, delete-orphan",
)
llm_configs = relationship(
"LLMConfig",
back_populates="user",
@ -331,6 +354,12 @@ else:
search_source_connectors = relationship(
"SearchSourceConnector", back_populates="user"
)
calendar_account = relationship(
"GoogleCalendarAccount",
back_populates="user",
uselist=False,
cascade="all, delete-orphan",
)
llm_configs = relationship(
"LLMConfig",
back_populates="user",