Skip to content

Commit 9347ac6

Browse files
committed
add DatabaseConstants
1 parent ddc7e76 commit 9347ac6

5 files changed

Lines changed: 17 additions & 22 deletions

File tree

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
from enum import Enum
2+
3+
class PaperSource(Enum):
4+
ARXIV = "ARXIV"
5+
6+
class PaperType(Enum):
7+
JOURNAL = "JOURNAL"
8+
CONFERENCE = "CONFERENCE"
9+
PREPRINT = "PREPRINT"
10+
WORKSHOP = "WORKSHOP"
11+
THESIS = "THESIS"
12+
OTHER = "OTHER"

backend/app/constants/__init__.py

Whitespace-only changes.

backend/app/models/paper.py

Lines changed: 2 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -3,30 +3,13 @@
33
# pylint: disable=too-few-public-methods
44

55
from datetime import datetime, date
6-
from enum import Enum
76
from typing import TYPE_CHECKING, List, Optional
87

98
from sqlalchemy import BigInteger, Date, DateTime, Enum as SqlEnum, JSON, String, Text, func
109
from sqlalchemy.orm import Mapped, mapped_column, relationship
1110

1211
from app.core.database import Base
13-
14-
15-
class PaperSource(str, Enum):
16-
"""Enumeration of supported paper sources."""
17-
18-
ARXIV = "Arxiv"
19-
20-
21-
class PaperType(str, Enum):
22-
"""Enumeration of supported paper types."""
23-
24-
JOURNAL = "Journal"
25-
CONFERENCE = "Conference"
26-
PREPRINT = "Preprint"
27-
WORKSHOP = "Workshop"
28-
THESIS = "Thesis"
29-
OTHER = "Other"
12+
from app.constants.DatabaseConstants import PaperSource, PaperType
3013

3114

3215
class Paper(Base):
@@ -62,6 +45,6 @@ class Paper(Base):
6245
"Project", secondary="project_paper", back_populates="papers"
6346
)
6447

65-
if TYPE_CHECKING: # pragma: no cover
48+
if TYPE_CHECKING:
6649
from .project import Project
6750
from .project_paper import ProjectPaper

backend/app/models/project.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ class Project(Base):
2020
created_by: Mapped[int] = mapped_column(
2121
BigInteger, ForeignKey("user.user_id"), nullable=False, index=True
2222
)
23-
project_name: Mapped[str] = mapped_column(String(255), nullable=False, unique=True)
23+
project_name: Mapped[str] = mapped_column(String(255), nullable=False)
2424
created_at: Mapped[datetime] = mapped_column(
2525
DateTime(timezone=True), nullable=False, server_default=func.now()
2626
)
@@ -33,7 +33,7 @@ class Project(Base):
3333
"Paper", secondary="project_paper", back_populates="projects"
3434
)
3535

36-
if TYPE_CHECKING: # pragma: no cover
36+
if TYPE_CHECKING:
3737
from .paper import Paper
3838
from .project_paper import ProjectPaper
3939
from .user import User

backend/app/models/project_paper.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,6 @@ class ProjectPaper(Base):
3232
paper: Mapped["Paper"] = relationship("Paper", back_populates="project_links")
3333

3434

35-
if TYPE_CHECKING: # pragma: no cover
35+
if TYPE_CHECKING:
3636
from .paper import Paper
3737
from .project import Project

0 commit comments

Comments
 (0)