i add tow fields in depertement

app/models/department.py

@@ -12,5 +12,7 @@ class Department(Base):
     code = Column(String(20), unique=True, nullable=False)
     description = Column(Text, nullable=True)
     keywords = Column(Text, nullable=True)
+    user_id = Column(Integer, nullable=True)
+    query = Column(Text, nullable=True)
     is_active = Column(Boolean, default=True, nullable=False)
     users = relationship("User", back_populates="department")

app/schemas/admin.py

@@ -6,6 +6,8 @@ class AdminDepartmentUpdate(BaseModel):
     code: str | None = Field(default=None, max_length=20)
     description: str | None = None
     keywords: str | None = None
+    user_id: int | None = None
+    query: str | None = None
     is_active: bool | None = None
 
 

app/schemas/department.py

@@ -6,6 +6,8 @@ class DepartmentCreate(BaseModel):
     code: str
     description: str | None = None
     keywords: str | None = None
+    user_id: int | None = None
+    query: str | None = None
 
 
 class DepartmentResponse(DepartmentCreate):

app/services/admin_service.py

@@ -40,6 +40,8 @@ def create_department(
     code: str,
     description: str | None = None,
     keywords: str | None = None,
+    user_id: int | None = None,
+    query: str | None = None,
 ) -> Department:
     if _department_exists_with_name_or_code(db, name=name):
         raise ValueError("Department name is already registered")
@@ -51,6 +53,8 @@ def create_department(
         code=code,
         description=description,
         keywords=keywords,
+        user_id=user_id,
+        query=query,
     )
     db.add(department)
     db.commit()
@@ -89,6 +93,10 @@ def update_department(
         department.description = payload.description
     if payload.keywords is not None:
         department.keywords = payload.keywords
+    if payload.user_id is not None:
+        department.user_id = payload.user_id
+    if payload.query is not None:
+        department.query = payload.query
     if payload.is_active is not None:
         department.is_active = payload.is_active
 

tests/test_admin.py

@@ -70,6 +70,38 @@ def test_admin_can_create_update_and_deactivate_department():
         Base.metadata.drop_all(bind=engine)
 
 
+def test_admin_can_store_user_and_query_on_department():
+    engine, db = _make_session()
+    try:
+        created = create_department(
+            db,
+            name="Student Support",
+            code="SSUP",
+            description="Student support desk",
+            keywords="support, help",
+            user_id=19,
+            query="Initial student support request",
+        )
+
+        assert created.user_id == 19
+        assert created.query == "Initial student support request"
+
+        updated = update_department(
+            db,
+            department_id=created.id,
+            payload=AdminDepartmentUpdate(
+                query="I need help with my fee payment",
+                is_active=True,
+            ),
+        )
+
+        assert updated.query == "I need help with my fee payment"
+        assert updated.user_id == 19
+    finally:
+        db.close()
+        Base.metadata.drop_all(bind=engine)
+
+
 def test_admin_can_list_and_update_user_department():
     engine, db = _make_session()
     try: