Develop

.env

@@ -3,9 +3,9 @@ DOMAIN=localhost
 # Configure these with your own Docker registry images
 DOCKER_IMAGE_BACKEND=backend
 DOCKER_IMAGE_CELERYWORKER=celerywork
+DOCKER_IMAGE_CELERYBEAT=celerybeat
 
 # Backend
-BACKEND_CORS_ORIGINS="[\"http://localhost\", \"http://localhost:4200\", \"http://localhost:3000\", \"http://localhost:8080\", \"https://localhost\", \"https://localhost:4200\", \"https://localhost:3000\", \"https://localhost:8080\", \"http://local.dockertoolbox.tiangolo.com\", \"http://localhost.tiangolo.com\"]"
 PROJECT_NAME="BeMore"
 SECRET_KEY=changethis
 SMTP_TLS=True

backend/.gitignore

@@ -2,3 +2,4 @@ __pycache__
 app.egg-info
 *.pyc
 coverage.xml
+*.pkl

backend/app/.gitignore

@@ -1,2 +1,3 @@
 .coverage
 htmlcov
+celerybeat-schedule

backend/app/app/conftest.py

@@ -8,15 +8,20 @@
 from app.web.application import get_app
 
 
-@pytest.fixture(name="session")
-def session_fixture():
+@pytest.fixture(scope="session")
+def engine_fixture():
     engine = create_engine(
-        f"postgresql+psycopg://{settings.POSTGRES_USER}:{settings.POSTGRES_PASSWORD}@bemore-db/{settings.POSTGRES_DB}",
+        f"postgresql+psycopg://{settings.POSTGRES_USER}:{settings.POSTGRES_PASSWORD}@bemore-db/test",
     )
     SQLModel.metadata.create_all(engine)
-
     with Session(engine) as session:
         init_db(session)
+    yield engine
+
+
+@pytest.fixture(name="session")
+def session_fixture(engine_fixture):
+    with Session(engine_fixture) as session:
         yield session
 
 

backend/app/app/core/config.py

@@ -61,10 +61,12 @@ class Settings(BaseSettings):
     SMTP_USER: str = ""
     SMTP_PASSWORD: str = ""
 
-    EMAIL_TEMPLATES_DIR: str = "bemore/email-templates/build"
+    EMAIL_TEMPLATES_DIR: str = "/app/app/email-templates/"
 
     # requests settings
     REQUESTS_BATCH_SIZE: int = 2
+    CRAWL_INTERVAL: int = 60 * 60 * 24 * 7  # 1 week
+    DATA_SIZE: int = 100
 
     model_config = SettingsConfigDict(
         env_file=".env",

backend/app/app/db/init_db.py

@@ -1,10 +1,17 @@
+import csv
 import logging
+import random
 
 from sqlmodel import Session, SQLModel, select
 
 from app.core.config import settings
 from app.db.engine import engine
-from app.models import User, UserCreate  # noqa: F401
+from app.models import (
+    FeedBack,
+    Item,
+    User,
+    UserCreate,
+)
 
 
 def init_db(session: Session) -> None:
@@ -20,6 +27,39 @@ def init_db(session: Session) -> None:
         user = User.create(session, user_in)
         logging.debug(f"Superuser {settings.FIRST_SUPERUSER} created")
 
+        # cold start for the recommender
+        users = [f"{i}@bemore.com" for i in range(1, 11)]
+        for email in users:
+            user_in = UserCreate(
+                email=email,
+                password="123456",
+                is_active=False,
+            )
+            user = User.create(session, user_in)
+
+        with open("/app/app/db/item.csv", "r") as file:
+            reader = csv.reader(file)
+            for i, row in enumerate(reader):
+                if i == 0:
+                    continue
+                item_in = Item(
+                    title=row[0],
+                    abstract=row[1],
+                    url=row[4],
+                    from_source=row[5],
+                )
+                item = Item.create(session, item_in)
+
+        for user_id in range(1, 11):
+            for i in range(1, 20):
+                feedback = FeedBack(
+                    user_id=user_id,
+                    item_id=random.randint(1, 100),
+                    feedback_type=1.0,
+                )
+                session.add(feedback)
+        session.commit()
+
 
 async def init():
     with Session(engine) as session: