pygpt-net 2.4.47__py3-none-any.whl

pygpt_net/provider/core/calendar/__init__.py

@@ -0,0 +1,10 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# ================================================== #
+# This file is a part of PYGPT package               #
+# Website: https://pygpt.net                         #
+# GitHub:  https://github.com/szczyglis-dev/py-gpt   #
+# MIT License                                        #
+# Created By  : Marcin Szczygliński                  #
+# Updated Date: 2023.12.31 04:00:00                  #
+# ================================================== #

pygpt_net/provider/core/calendar/base.py

@@ -0,0 +1,56 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# ================================================== #
+# This file is a part of PYGPT package               #
+# Website: https://pygpt.net                         #
+# GitHub:  https://github.com/szczyglis-dev/py-gpt   #
+# MIT License                                        #
+# Created By  : Marcin Szczygliński                  #
+# Updated Date: 2024.12.14 22:00:00                  #
+# ================================================== #
+
+from typing import Dict
+
+from packaging.version import Version
+
+from pygpt_net.item.calendar_note import CalendarNoteItem
+
+
+class BaseProvider:
+    def __init__(self, window=None):
+        self.window = window
+        self.id = ""
+        self.type = "calendar_note"
+
+    def attach(self, window):
+        self.window = window
+
+    def install(self):
+        pass
+
+    def patch(self, version: Version) -> bool:
+        pass
+
+    def create(self, notepad: CalendarNoteItem):
+        pass
+
+    def load(self, year, month, day) -> CalendarNoteItem:
+        pass
+
+    def load_all(self) -> Dict[str, CalendarNoteItem]:
+        pass
+
+    def save(self, notepad: CalendarNoteItem):
+        pass
+
+    def save_all(self, items: Dict[str, CalendarNoteItem]):
+        pass
+
+    def remove(self, year, month, day):
+        pass
+
+    def truncate(self):
+        pass
+
+    def get_version(self) -> str:
+        pass

pygpt_net/provider/core/calendar/db_sqlite/__init__.py

@@ -0,0 +1,150 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# ================================================== #
+# This file is a part of PYGPT package               #
+# Website: https://pygpt.net                         #
+# GitHub:  https://github.com/szczyglis-dev/py-gpt   #
+# MIT License                                        #
+# Created By  : Marcin Szczygliński                  #
+# Updated Date: 2024.12.14 22:00:00                  #
+# ================================================== #
+
+import uuid
+from typing import Dict
+
+from packaging.version import Version
+
+from pygpt_net.item.calendar_note import CalendarNoteItem
+from pygpt_net.provider.core.calendar.base import BaseProvider
+from .patch import Patch
+from .storage import Storage
+
+
+class DbSqliteProvider(BaseProvider):
+    def __init__(self, window=None):
+        super(DbSqliteProvider, self).__init__(window)
+        self.window = window
+        self.patcher = Patch(window)
+        self.storage = Storage(window)
+        self.id = "db_sqlite"
+        self.type = "calendar_note"
+
+    def attach(self, window):
+        """
+        Attach window instance to provider
+
+        :param window: window instance
+        """
+        self.window = window
+        self.storage.attach(window)
+
+    def patch(self, version: Version) -> bool:
+        """
+        Patch versions
+
+        :param version: current app version
+        :return: True if migrated
+        """
+        return self.patcher.execute(version)
+
+    def create_id(self) -> str:
+        """
+        Create unique uuid
+
+        :return: uuid
+        """
+        return str(uuid.uuid4())
+
+    def create(self, note: CalendarNoteItem) -> int:
+        """
+        Create new and return its ID
+
+        :param note: CalendarNoteItem
+        :return: note ID
+        """
+        if note.id is None or note.id == "":
+            note.id = self.storage.insert(note)
+        return note.id
+
+    def load_all(self) -> Dict[str, CalendarNoteItem]:
+        """
+        Load notes from DB
+
+        :return: dict with notes
+        """
+        return self.storage.get_all()
+
+    def load_by_month(self, year: int, month: int) -> Dict[str, CalendarNoteItem]:
+        """
+        Load notes from DB
+
+        :return: notes
+        """
+        return self.storage.get_by_month(year, month)
+
+    def load(self, year: int, month: int, day: int) -> CalendarNoteItem:
+        """
+        Load note from DB
+
+        :param year: year
+        :param month: month
+        :param day: day
+        :return: CalendarNoteItem object
+        """
+        return self.storage.get_by_date(year, month, day)
+
+    def get_notes_existence_by_day(self, year, month) -> Dict[str, Dict[int, int]]:
+        """
+        Get notes existence by day
+
+        :param year: year
+        :param month: month
+        :return: notes existence by day
+        """
+        return self.storage.get_notes_existence_by_day(year, month)
+
+    def save(self, note: CalendarNoteItem):
+        """
+        Save note to DB
+
+        :param note: CalendarNoteItem
+        """
+        try:
+            self.storage.save(note)
+        except Exception as e:
+            self.window.core.debug.log(e)
+            print("Error while saving note: {}".format(str(e)))
+
+    def save_all(self, items: Dict[str, CalendarNoteItem]):
+        """
+        Save all notes to DB
+
+        :param items: dict of CalendarNoteItem objects
+        """
+        try:
+            for dt in items:
+                note = items[dt]
+                self.storage.save(note)
+        except Exception as e:
+            self.window.core.debug.log(e)
+            print("Error while saving note: {}".format(str(e)))
+
+    def remove(self, year: int, month: int, day: int):
+        """
+        Remove note from DB
+
+        :param year: year
+        :param month: month
+        :param day: day
+        """
+        self.storage.delete_by_date(year, month, day)
+
+    def truncate(self) -> bool:
+        """
+        Truncate all notes
+
+        :return: True if truncated
+        :rtype: bool
+        """
+        return self.storage.truncate_all()
+

pygpt_net/provider/core/calendar/db_sqlite/patch.py

@@ -0,0 +1,26 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# ================================================== #
+# This file is a part of PYGPT package               #
+# Website: https://pygpt.net                         #
+# GitHub:  https://github.com/szczyglis-dev/py-gpt   #
+# MIT License                                        #
+# Created By  : Marcin Szczygliński                  #
+# Updated Date: 2024.01.12 04:00:00                  #
+# ================================================== #
+
+from packaging.version import parse as parse_version, Version
+
+
+class Patch:
+    def __init__(self, window=None):
+        self.window = window
+
+    def execute(self, version: Version) -> bool:
+        """
+        Migrate to current app version
+
+        :param version: current app version
+        :return: True if migrated
+        """
+        pass

pygpt_net/provider/core/calendar/db_sqlite/storage.py

@@ -0,0 +1,328 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# ================================================== #
+# This file is a part of PYGPT package               #
+# Website: https://pygpt.net                         #
+# GitHub:  https://github.com/szczyglis-dev/py-gpt   #
+# MIT License                                        #
+# Created By  : Marcin Szczygliński                  #
+# Updated Date: 2024.12.14 22:00:00                  #
+# ================================================== #
+
+import time
+from typing import Dict
+
+from sqlalchemy import text
+
+from pygpt_net.item.calendar_note import CalendarNoteItem
+from pygpt_net.utils import unpack_var
+
+
+class Storage:
+    def __init__(self, window=None):
+        """
+        Initialize storage instance
+
+        :param window: Window instance
+        """
+        self.window = window
+
+    def attach(self, window):
+        """
+        Attach window instance
+
+        :param window: Window instance
+        """
+        self.window = window
+
+    def get_all(self) -> Dict[str, CalendarNoteItem]:
+        """
+        Return dict with CalendarNoteItem objects, indexed by YYYY-MM-DD
+
+        :return: dict of CalendarNoteItem objects
+        """
+        stmt = text("""
+            SELECT * FROM calendar_note
+        """)
+        items = {}
+        db = self.window.core.db.get_db()
+        with db.connect() as conn:
+            result = conn.execute(stmt)
+            for row in result:
+                note = CalendarNoteItem()
+                self.unpack(note, row._asdict())
+                dt = note.get_dt()
+                items[dt] = note
+        return items
+
+    def get_by_month(self, year: int, month: int) -> Dict[str, CalendarNoteItem]:
+        """
+        Return dict with CalendarNoteItem objects, indexed by YYYY-MM-DD
+
+        :param year: year
+        :param month: month
+        :return: dict of CalendarNoteItem objects
+        """
+        stmt = text("""
+            SELECT * FROM calendar_note WHERE year = :year AND month = :month
+        """).bindparams(year=year, month=month)
+        items = {}
+        db = self.window.core.db.get_db()
+        with db.connect() as conn:
+            result = conn.execute(stmt)
+            for row in result:
+                note = CalendarNoteItem()
+                self.unpack(note, row._asdict())
+                dt = note.get_dt()
+                items[dt] = note
+        return items
+
+    def get_by_date(self, year: int, month: int, day: int) -> CalendarNoteItem:
+        """
+        Return CalendarNoteItem by date
+
+        :param year: year
+        :param month: month
+        :param day: day
+        :return: CalendarNoteItem
+        """
+        stmt = text("""
+            SELECT * FROM calendar_note WHERE year = :year AND month = :month AND day = :day LIMIT 1
+        """).bindparams(year=year, month=month, day=day)
+        notepad = CalendarNoteItem()
+        db = self.window.core.db.get_db()
+        with db.connect() as conn:
+            result = conn.execute(stmt)
+            for row in result:
+                self.unpack(notepad, row._asdict())
+        return notepad
+
+    def get_notes_existence_by_day(self, year: int, month: int) -> Dict[str, Dict[int, int]]:
+        """
+        Return a dict of days with the count of notes for the given year and month.
+
+        :param year: year
+        :param month: month
+        :return: dict of days with the count of notes
+        """
+        db = self.window.core.db.get_db()
+        with db.connect() as conn:
+            result = conn.execute(text("""
+                    SELECT
+                        year || '-' || printf('%02d', month) || '-' || printf('%02d', day) as day,
+                        status,
+                        COUNT(id) as note_count
+                    FROM calendar_note
+                    WHERE year = :year
+                      AND month = :month
+                      AND is_deleted = 0
+                      AND content != ''
+                    GROUP BY day, status
+                """), {'year': year, 'month': month})
+
+            days_with_notes = {}
+            for row in result:
+                day = row._mapping['day']
+                status = row._mapping['status']
+                note_count = row._mapping['note_count']
+
+                if day not in days_with_notes:
+                    days_with_notes[day] = {}
+
+                if note_count > 0:
+                    days_with_notes[day][status] = note_count
+
+            return days_with_notes
+
+    def truncate_all(self) -> bool:
+        """
+        Truncate all notes from database
+
+        :return: True if truncated
+        """
+        db = self.window.core.db.get_db()
+        with db.begin() as conn:
+            conn.execute(text("DELETE FROM calendar_note"))
+        return True
+
+    def delete_by_date(self, year: int, month: int, day: int) -> bool:
+        """
+        Delete note item by date
+
+        :param year: year
+        :param month: month
+        :param day: day
+        :return: True if deleted
+        """
+        stmt = text("""
+            DELETE FROM calendar_note WHERE year = :year AND month = :month AND day = :day
+        """).bindparams(year=year, month=month, day=day)
+        db = self.window.core.db.get_db()
+        with db.begin() as conn:
+            conn.execute(stmt)
+            return True
+
+    def save(self, note: CalendarNoteItem):
+        """
+        Insert or update note item
+
+        :param note: CalendarNoteItem object
+        """
+        db = self.window.core.db.get_db()
+        with db.begin() as conn:
+            ts = int(time.time())
+            sel_stmt = text("SELECT 1 FROM calendar_note WHERE year = :year AND month = :month AND day = :day LIMIT 1").bindparams(
+                year=note.year,
+                month=note.month,
+                day=note.day
+            )
+            result = conn.execute(sel_stmt).fetchone()
+            if result:
+                stmt = text("""
+                            UPDATE calendar_note
+                            SET
+                                title = :title,
+                                status = :status,
+                                content = :content,
+                                is_important = :is_important,
+                                updated_ts = :updated_ts
+                            WHERE year = :year AND month = :month AND day = :day
+                        """).bindparams(
+                    title=note.title,
+                    year=int(note.year),
+                    month=int(note.month),
+                    day=int(note.day),
+                    status=note.status,
+                    content=note.content,
+                    is_important=int(note.important),
+                    updated_ts=ts
+                )
+            else:
+                stmt = text("""
+                            INSERT INTO calendar_note 
+                            (
+                                idx,
+                                uuid,
+                                status,
+                                year,
+                                month,
+                                day,
+                                title, 
+                                content, 
+                                created_ts, 
+                                updated_ts,
+                                is_important,
+                                is_deleted
+                            )
+                            VALUES
+                            (
+                                :idx,
+                                :uuid,
+                                :status,
+                                :year,
+                                :month,
+                                :day,
+                                :title,
+                                :content,
+                                :created_ts,
+                                :updated_ts,
+                                :is_important,
+                                :is_deleted
+                            )
+                        """).bindparams(
+                    idx=note.idx,
+                    uuid=note.uuid,
+                    status=int(note.status),
+                    year=int(note.year),
+                    month=int(note.month),
+                    day=int(note.day),
+                    title=note.title,
+                    content=note.content,
+                    created_ts=ts,
+                    updated_ts=ts,
+                    is_important=int(note.important),
+                    is_deleted=int(note.deleted)
+                )
+            conn.execute(stmt)
+
+    def insert(self, note: CalendarNoteItem) -> int:
+        """
+        Insert note item
+
+        :param note: CalendarNoteItem object
+        :return: note item ID
+        """
+        db = self.window.core.db.get_db()
+        ts = int(time.time())
+        stmt = text("""
+                INSERT INTO calendar_note 
+                (
+                    idx,
+                    uuid,
+                    status,
+                    year,
+                    month,
+                    day,
+                    title, 
+                    content, 
+                    created_ts, 
+                    updated_ts,
+                    is_important,
+                    is_deleted
+                )
+                VALUES
+                (
+                    :idx,
+                    :uuid,
+                    :status,
+                    :year,
+                    :month,
+                    :day,
+                    :title,
+                    :content,
+                    :created_ts,
+                    :updated_ts,
+                    :is_important,
+                    :is_deleted
+                )
+            """).bindparams(
+            idx=int(note.idx or 0),
+            uuid=note.uuid,
+            status=int(note.status or 0),
+            year=int(note.year or 0),
+            month=int(note.month or 0),
+            day=int(note.day or 0),
+            title=note.title,
+            content=note.content,
+            created_ts=ts,
+            updated_ts=ts,
+            is_important=int(note.important),
+            is_deleted=int(note.deleted)
+        )
+        with db.begin() as conn:
+            result = conn.execute(stmt)
+            note.id = result.lastrowid
+            return note.id
+
+    def unpack(self, note: CalendarNoteItem, row: dict) -> CalendarNoteItem:
+        """
+        Unpack note item from DB row
+
+        :param note: CalendarNoteItem item
+        :param row: DB row
+        :return: CalendarNoteItem item
+        """
+        note.id = unpack_var(row['id'], 'int')
+        note.idx = unpack_var(row['idx'], 'int')
+        note.uuid = row['uuid']
+        note.status = unpack_var(row['status'], 'int')
+        note.year = unpack_var(row['year'], 'int')
+        note.month = unpack_var(row['month'], 'int')
+        note.day = unpack_var(row['day'], 'int')
+        note.created = unpack_var(row['created_ts'], 'int')
+        note.updated = unpack_var(row['updated_ts'], 'int')
+        note.title = row['title']
+        note.content = row['content']
+        note.important = unpack_var(row['is_important'], 'bool')
+        note.deleted = unpack_var(row['is_deleted'], 'bool')
+        return note

pygpt_net/provider/core/config/__init__.py

@@ -0,0 +1,10 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# ================================================== #
+# This file is a part of PYGPT package               #
+# Website: https://pygpt.net                         #
+# GitHub:  https://github.com/szczyglis-dev/py-gpt   #
+# MIT License                                        #
+# Created By  : Marcin Szczygliński                  #
+# Updated Date: 2023.12.31 04:00:00                  #
+# ================================================== #

pygpt_net/provider/core/config/base.py

@@ -0,0 +1,51 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# ================================================== #
+# This file is a part of PYGPT package               #
+# Website: https://pygpt.net                         #
+# GitHub:  https://github.com/szczyglis-dev/py-gpt   #
+# MIT License                                        #
+# Created By  : Marcin Szczygliński                  #
+# Updated Date: 2024.12.14 22:00:00                  #
+# ================================================== #
+
+from typing import Dict, Any, Optional
+
+from packaging.version import Version
+
+
+class BaseProvider:
+    def __init__(self, window=None):
+        self.window = window
+        self.id = ""
+        self.type = "config"
+        self.path = None
+        self.path_app = None
+        self.meta = None
+
+    def attach(self, window):
+        self.window = window
+
+    def install(self):
+        pass
+
+    def patch(self, version: Version) -> bool:
+        pass
+
+    def load(self, all: bool = False) -> Optional[Dict[str, Any]]:
+        pass
+
+    def load_base(self) -> Optional[Dict[str, Any]]:
+        pass
+
+    def save(self, items: Dict[str, Any], filename: str = 'config.json'):
+        pass
+
+    def get_options(self) -> Optional[Dict[str, Any]]:
+        pass
+
+    def get_sections(self) -> Optional[Dict[str, Any]]:
+        pass
+
+    def get_version(self) -> str:
+        pass