Compare commits

...

9 Commits

2 changed files with 112 additions and 14 deletions

View File

@ -1,18 +1,99 @@
class History: #Abstract class
def __init__(self, filename):
self.history = list()
def __size__(self):
return len(history)
def parse_history(self, filename):
return history
#
# 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/>.
#
def is_this_type(self, filename): # bool function,
return false # returns false if Youtube history
def get_video(self, index):
return history[index]
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):
return iter(history)
for entry in self._parsed_data:
yield FreeTubeHistory._to_video(entry)

View File

@ -1,3 +1,20 @@
#
# 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,