Implement duty schedule import functionality and enhance user management

- Added a new command `/import_duty_schedule` for importing duty schedules via JSON, restricted to admin users.
- Introduced a two-step import process: specifying handover time and uploading a JSON file.
- Updated the database schema to allow `telegram_user_id` to be nullable for user creation by full name.
- Implemented repository functions for user management, including `get_or_create_user_by_full_name` and `delete_duties_in_range`.
- Enhanced README documentation with details on the new import command and JSON format requirements.
- Added comprehensive tests for the duty schedule parser and integration tests for the import functionality.

db/__init__.py

@@ -3,7 +3,13 @@
 from db.models import Base, User, Duty
 from db.schemas import UserCreate, UserInDb, DutyCreate, DutyInDb, DutyWithUser
 from db.session import get_engine, get_session_factory, get_session
-from db.repository import get_or_create_user, get_duties, insert_duty
+from db.repository import (
+    delete_duties_in_range,
+    get_or_create_user,
+    get_or_create_user_by_full_name,
+    get_duties,
+    insert_duty,
+)
 
 __all__ = [
     "Base",
@@ -17,7 +23,9 @@ __all__ = [
     "get_engine",
     "get_session_factory",
     "get_session",
+    "delete_duties_in_range",
     "get_or_create_user",
+    "get_or_create_user_by_full_name",
     "get_duties",
     "insert_duty",
     "init_db",

db/models.py

@@ -14,8 +14,8 @@ class User(Base):
     __tablename__ = "users"
 
     id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
-    telegram_user_id: Mapped[int] = mapped_column(
-        BigInteger, unique=True, nullable=False
+    telegram_user_id: Mapped[int | None] = mapped_column(
+        BigInteger, unique=True, nullable=True
     )
     full_name: Mapped[str] = mapped_column(Text, nullable=False)
     username: Mapped[str | None] = mapped_column(Text, nullable=True)

db/repository.py

@@ -1,5 +1,7 @@
 """Repository: get_or_create_user, get_duties, insert_duty."""
 
+from datetime import datetime, timedelta
+
 from sqlalchemy.orm import Session
 
 from db.models import User, Duty
@@ -35,6 +37,47 @@ def get_or_create_user(
     return user
 
 
+def get_or_create_user_by_full_name(session: Session, full_name: str) -> User:
+    """Find user by exact full_name or create one with telegram_user_id=None (for duty-schedule import)."""
+    user = session.query(User).filter(User.full_name == full_name).first()
+    if user:
+        return user
+    user = User(
+        telegram_user_id=None,
+        full_name=full_name,
+        username=None,
+        first_name=None,
+        last_name=None,
+    )
+    session.add(user)
+    session.commit()
+    session.refresh(user)
+    return user
+
+
+def delete_duties_in_range(
+    session: Session,
+    user_id: int,
+    from_date: str,
+    to_date: str,
+) -> int:
+    """Delete all duties of the user that overlap [from_date, to_date] (YYYY-MM-DD). Returns count deleted."""
+    # start_at < to_date + 1 day so duties starting on to_date are included (start_at is ISO with T)
+    to_next = (datetime.fromisoformat(to_date + "T00:00:00") + timedelta(days=1)).strftime("%Y-%m-%d")
+    q = (
+        session.query(Duty)
+        .filter(
+            Duty.user_id == user_id,
+            Duty.start_at < to_next,
+            Duty.end_at >= from_date,
+        )
+    )
+    count = q.count()
+    q.delete(synchronize_session=False)
+    session.commit()
+    return count
+
+
 def get_duties(
     session: Session,
     from_date: str,