Добавить core

core

@@ -0,0 +1,103 @@
+from datetime import date, datetime
+
+
+class HarmCalculator:
+
+    harms = {
+        "Никотин": {"сигарета": 11, "стик": 5, "тяжка": 3},
+        "Алкоголь": {"бутылка (500мл)": 15, "банка (500мл)": 12, "бокал (150мл)": 8, "рюмка (50мл)": 28, "глоток (30мл)": 4, "порция (250мл)": 18},
+        "Кофе": {"чашка (200мл)": 1, "стакан (350мл)": 1.5, "глоток (30мл)": 0.2, "порция (200мл)": 1},
+        "Энергетики": {"банка (250мл)": 2, "банка (330мл)": 3, "банка (500мл)": 5},
+    }
+
+    icons = {"Никотин": "🚬", "Алкоголь": "🍷", "Кофе": "☕", "Энергетики": "⚡"}
+    colors = {"Никотин": "#8fb0d9", "Алкоголь": "#f0a2bc", "Кофе": "#c8a07a", "Энергетики": "#7ea8ff"}
+    gender_mult = {"Мужской": 1.0, "Женский": 1.2, "М": 1.0, "Ж": 1.2}
+
+    def __init__(self, gender="Мужской", age=20):
+        self.gender = gender
+        self.age = age
+
+    def set_profile(self, gender, age):
+        self.gender = gender
+        self.age = age
+
+    def age_mult(self, age):
+        if age < 18: return 0.9
+        if age < 25: return 0.95
+        if age < 40: return 1.0
+        if age < 60: return 1.2
+        return 1.5
+
+    def calc_one(self, kind, unit, qty):
+        base = self.harms[kind][unit]
+        g = self.gender_mult.get(self.gender, 1.0)
+        a = self.age_mult(self.age)
+        return base * qty * g * a
+
+    def calc_total(self, hist):
+        total = 0
+        for e in hist:
+            total += self.calc_one(e["kind"], e["unit"], e["qty"])
+        return total
+
+    def fmt_time(self, mins):
+        if mins < 60: return f"{int(mins)} мин"
+        if mins < 1440: return f"{mins/60:.1f} ч"
+        if mins < 43200: return f"{mins/1440:.2f} дн"
+        return f"{mins/43200:.2f} мес"
+
+    def fmt_timer(self, mins):
+        h = int(mins // 60)
+        m = int(mins % 60)
+        s = int((mins * 60) % 60)
+        return f"{h:02d}:{m:02d}:{s:02d}"
+
+    def get_units(self, kind):
+        return list(self.harms.get(kind, {}).keys())
+
+    def get_config(self, kind):
+        return {
+            "icon": self.icons.get(kind, "📌"),
+            "color": self.colors.get(kind, "#cccccc"),
+            "units": self.get_units(kind),
+        }
+
+    def period_harm(self, hist, period="day"):
+        today = date.today()
+        filt = []
+        for e in hist:
+            d = datetime.fromisoformat(e["date"]).date()
+            if period == "day" and d == today:
+                filt.append(e)
+            elif period == "month" and d.year == today.year and d.month == today.month:
+                filt.append(e)
+            elif period == "year" and d.year == today.year:
+                filt.append(e)
+            elif period == "all":
+                filt.append(e)
+        return self.calc_total(filt), len(filt)
+
+    def days_with_harms(self, hist, year, month):
+        days = {}
+        for e in hist:
+            d = datetime.fromisoformat(e["date"]).date()
+            if d.year == year and d.month == month:
+                h = self.calc_one(e["kind"], e["unit"], e["qty"])
+                days[d.day] = days.get(d.day, 0) + h
+        return days
+
+    def summary(self, hist):
+        day = self.period_harm(hist, "day")
+        month = self.period_harm(hist, "month")
+        year = self.period_harm(hist, "year")
+        allt = self.period_harm(hist, "all")
+        return {
+            "day": {"mins": day[0], "fmt": self.fmt_time(day[0]), "cnt": day[1]},
+            "month": {"mins": month[0], "fmt": self.fmt_time(month[0]), "cnt": month[1]},
+            "year": {"mins": year[0], "fmt": self.fmt_time(year[0]), "cnt": year[1]},
+            "all": {"mins": allt[0], "fmt": self.fmt_time(allt[0]), "cnt": allt[1]},
+        }
+
+    def to_days(self, mins):
+        return mins / 1440