Compare commits
No commits in common. "main" and "License_headers" have entirely different histories.
main
...
License_he
@ -1,99 +1,18 @@
|
|||||||
#
|
class History: #Abstract class
|
||||||
# Copyright (c) 2025 Mykola Shulhin.
|
def __init__(self, filename):
|
||||||
# Copyright (c) 2025 Fedir Kovalov.
|
self.history = list()
|
||||||
#
|
|
||||||
# This program is free software: you can redistribute it and/or modify
|
def __size__(self):
|
||||||
# it under the terms of the GNU General Public License as published by
|
return len(history)
|
||||||
# the Free Software Foundation, version 3.
|
|
||||||
#
|
def parse_history(self, filename):
|
||||||
# This program is distributed in the hope that it will be useful, but
|
return history
|
||||||
# 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
|
||||||
|
|
||||||
import json
|
def get_video(self, index):
|
||||||
from typing import Iterator
|
return history[index]
|
||||||
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):
|
def __iter__(self):
|
||||||
for entry in self._parsed_data:
|
return iter(history)
|
||||||
yield FreeTubeHistory._to_video(entry)
|
|
||||||
|
Loading…
Reference in New Issue
Block a user