machinelearning/models.py at main · AishaRiyad/machinelearning · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
from sqlalchemy import Column, String, Text
from sqlalchemy.orm import Mapped, mapped_column
from database import Base
import json

class User(Base):
    __tablename__ = "users"
    id: Mapped[str] = mapped_column(String, primary_key=True)
    name: Mapped[str] = mapped_column(String, nullable=False)
    email: Mapped[str] = mapped_column(String, unique=True, index=True, nullable=False)
    pass_hash: Mapped[str] = mapped_column(String, nullable=False)

class Assessment(Base):
    __tablename__ = "assessments"
    id: Mapped[str] = mapped_column(String, primary_key=True)
    user_id: Mapped[str] = mapped_column(String, index=True, nullable=False)
    scores_json: Mapped[str] = mapped_column(Text, default="{}")
    signals_json: Mapped[str] = mapped_column(Text, default="{}")
    created_at: Mapped[str] = mapped_column(String, nullable=False)

    def set_scores(self, obj): self.scores_json = json.dumps(obj or {}, ensure_ascii=False)
    def get_scores(self):
        try: return json.loads(self.scores_json or "{}")
        except: return {}
    def set_signals(self, obj): self.signals_json = json.dumps(obj or {}, ensure_ascii=False)
    def get_signals(self):
        try: return json.loads(self.signals_json or "{}")
        except: return {}

class Evaluation(Base):
    __tablename__ = "evaluations"
    id: Mapped[str] = mapped_column(String, primary_key=True)
    user_id: Mapped[str] = mapped_column(String, index=True, nullable=False)
    assessment_id: Mapped[str] = mapped_column(String, nullable=False)
    domain_scores_json: Mapped[str] = mapped_column(Text, default="{}")
    created_at: Mapped[str] = mapped_column(String, nullable=False)

    def set_domain_scores(self, obj): self.domain_scores_json = json.dumps(obj or {}, ensure_ascii=False)
    def get_domain_scores(self):
        try: return json.loads(self.domain_scores_json or "{}")
        except: return {}

class Plan(Base):
    __tablename__ = "plans"
    id: Mapped[str] = mapped_column(String, primary_key=True)
    user_id: Mapped[str] = mapped_column(String, index=True, nullable=False)
    evaluation_id: Mapped[str] = mapped_column(String, nullable=False)
    items_json: Mapped[str] = mapped_column(Text, default="[]")
    created_at: Mapped[str] = mapped_column(String, nullable=False)
    started_at: Mapped[str] = mapped_column(String, nullable=True)
    # ⇦ جديد: نخزن نصائح القواعد
    advice_json: Mapped[str] = mapped_column(Text, default="[]")

    def set_items(self, arr): self.items_json = json.dumps(arr or [], ensure_ascii=False)
    def get_items(self):
        try: return json.loads(self.items_json or "[]")
        except: return []

    def set_advice(self, arr): self.advice_json = json.dumps(arr or [], ensure_ascii=False)
    def get_advice(self):
        try: return json.loads(self.advice_json or "[]")
        except: return []