Added a license header to history.

Reviewed-on: #13

shadowtube/history.py

@@ -0,0 +1,99 @@
+#
+# Copyright (c) 2025 Mykola Shulhin.
+# Copyright (c) 2025 Fedir Kovalov.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, version 3.
+#
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+#
+
+
+import json
+from typing import Iterator
+from .types import Video
+from abc import ABC, abstractmethod
+
+
+class History(ABC):  # Abstract class
+    @abstractmethod
+    def __len__(self) -> int:
+        pass
+
+    @staticmethod
+    def parse_history(filename: str) -> "History":
+        return FreeTubeHistory(filename)
+
+    @abstractmethod
+    def is_this_type(self, filename: str) -> bool:
+        pass
+
+    @abstractmethod
+    def get_video(self, index: int) -> Video:
+        pass
+
+    @abstractmethod
+    def __iter__(self) -> Iterator[Video]:
+        pass
+
+
+class FreeTubeHistory(History):
+    def __init__(self, filename: str) -> None:
+        parsed_data = []
+
+        with open(filename, "r", encoding="utf-8") as file:
+            for line in file:
+                line = line.strip()
+                if not line:
+                    continue
+                try:
+                    parsed_data.append(json.loads(line))
+                except json.JSONDecodeError:
+                    fixed_line = FreeTubeHistory._fix_unquoted_values(line)
+                    parsed_data.append(json.loads(fixed_line))
+
+        self._parsed_data = parsed_data
+
+    @staticmethod
+    def _fix_unquoted_values(line: str) -> str:
+        """Attempts to fix unquoted values by adding quotes around them."""
+        import re
+
+        def replacer(match):
+            key, value = match.groups()
+            if not (value.startswith('"') and value.endswith('"')):
+                value = f'"{value}"'  # Add quotes around the value
+            return f'"{key}":{value}'
+
+        fixed_line = re.sub(r'"(\w+)":(\w+)', replacer, line)
+        return fixed_line
+
+    @staticmethod
+    def _to_video(entry) -> Video:
+        return Video(
+            id=entry["videoId"],
+            title=entry["title"],
+            description=entry["description"],
+            watch_time=entry["timeWatched"],
+            watch_progress=entry["watchProgress"],
+        )
+
+    def __len__(self):
+        return len(self._parsed_data)
+
+    def is_this_type(self, filename: str) -> bool:
+        raise NotImplementedError()
+
+    def get_video(self, index: int) -> Video:
+        return FreeTubeHistory._to_video(self._parsed_data[index])
+
+    def __iter__(self):
+        for entry in self._parsed_data:
+            yield FreeTubeHistory._to_video(entry)

shadowtube/types.py

@@ -0,0 +1,51 @@
+#
+# Copyright (c) 2025 Fedir Kovalov.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, version 3.
+#
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+#
+
+
+class Video:
+    def __init__(
+        self,
+        id: str,
+        title: str,
+        description: str,
+        watch_time: float,
+        watch_progress: float,
+    ):
+        self._id = id
+        self._title = title
+        self._description = description
+        self._watch_time = watch_time
+        self._watch_progress = watch_progress
+
+    @property
+    def title(self):
+        return self._title
+
+    @property
+    def id(self):
+        return self._id
+
+    @property
+    def description(self):
+        return self._description
+
+    @property
+    def watch_time(self):
+        return self._watch_time
+
+    @property
+    def watch_progress(self):
+        return self._watch_progress