shifter-converter 0.0.1__tar.gz

shifter_converter-0.0.1/PKG-INFO

@@ -0,0 +1,19 @@
+Metadata-Version: 2.4
+Name: shifter-converter
+Version: 0.0.1
+Summary: A program for converting .Shifter files into .ics files
+Description-Content-Type: text/markdown
+Requires-Dist: ics>=0.7.3
+Requires-Dist: tzlocal>=5.3.1
+Requires-Dist: rich>=15.0.0
+
+# Shifter-Converter
+
+A program to convert .Shifter files to .ics files
+
+## AI
+
+AI was used for the following:
+
+- Formatting output of database commands
+- Instructions on setting up PyPI project

shifter_converter-0.0.1/README.md

@@ -0,0 +1,10 @@
+# Shifter-Converter
+
+A program to convert .Shifter files to .ics files
+
+## AI
+
+AI was used for the following:
+
+- Formatting output of database commands
+- Instructions on setting up PyPI project

shifter_converter-0.0.1/pyproject.toml

@@ -0,0 +1,20 @@
+[build-system]
+requires = ["setuptools >= 77.0.3"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "shifter-converter"
+version = "0.0.1"
+description = "A program for converting .Shifter files into .ics files"
+readme = "README.md"
+dependencies = [
+    "ics>=0.7.3",
+    "tzlocal>=5.3.1",
+    "rich>=15.0.0"
+]
+
+[project.scripts]
+shifter-converter = "shifter_converter.cli:main"
+
+[tool.setuptools.packages.find]
+where = ["src"]

shifter_converter-0.0.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

shifter_converter-0.0.1/src/shifter_converter/cli.py

@@ -0,0 +1,59 @@
+from rich.console import Console
+from rich.prompt import Prompt
+from rich.panel import Panel
+import sqlite3
+from .converter import main as converter_main
+import questionary
+import os
+from .converter import validate_shifter_file
+
+cwd = os.getcwd()
+console = Console()
+
+
+def get_file_path(name, folder=False):
+    return questionary.path(f"Give file path of {name}", default=cwd, only_directories=folder).ask()
+
+
+def validate_file(file_path: str):
+    try:
+        db = sqlite3.connect(file_path)
+        if validate_shifter_file(db):
+            console.print("[green]File is a valid .Shifter file[/green]")
+            return db
+        else:
+            console.print("[yellow]File is a valid sqlite3 database, but doesn't contain the correct tables for a .Shifter file[/yellow]")
+            main()
+    except Exception as e:
+        console.print(f"[red]File is not a valid sqlite3 database: {e}[/red]")
+        main()
+
+
+def main():
+    console.print(Panel.fit("Shifter Converter", style=" bold"))
+    console.print("Main menu:")
+    console.print("[1] Quick convert (.Shifter -> .ics)")
+    console.print("[2] Validate .Shifter")
+    console.print("[3] Exit")
+    option = Prompt.ask("Select option:", choices=["1", "2", "3", "4"], default="1")
+
+    if option == "1":
+        file_path = get_file_path(".Shifter file")
+        shifter_file = validate_file(file_path)
+        output_dir = get_file_path("output directory", folder=True)
+        converter_main(shifter_file, output_dir)  # pyright: ignore[reportArgumentType]
+        console.print("[green]Succesfully converted .Shifter to .ics[/green]")
+        console.print(f"[cyan]Saved as [italic]{os.path.join(output_dir, 'output_calendar.ics')}[/italic][/cyan]")
+        main()
+    elif option == "2":
+        file_path = get_file_path(".Shifter file")
+        validate_file(file_path)
+        main()
+    elif option == "3":
+        quit()
+    else:
+        print("No clue how you did it, but you selected a number outside of the normal range")
+
+
+if __name__ == "__main__":
+    main()

shifter_converter-0.0.1/src/shifter_converter/converter.py

@@ -0,0 +1,230 @@
+from sqlite3 import Connection
+from typing import Any, Optional
+from datetime import datetime, date, time, timedelta
+from dataclasses import dataclass, asdict
+import ics
+import re
+from tzlocal import get_localzone
+import os
+
+shift_lables = ["_id", "text", "alarmTime", "alarm", "alarmOneDayBefore", "color", "textColor", "sequentialCode", "textSize", "alarmTime2", "alarm2", "alarmTwoDayBefore", "sound1", "sound2", "abbreviation", "startTime1", "endTime1", "startTime2", "endTime2", "splitShift", "performAction", "hourlyEarnings", "extraHourlyEarnings", "restTime", "shiftTime", "calculatedManualShiftDuration"]
+
+local_tz = get_localzone()
+
+
+@dataclass
+class Shift():
+    _id: int
+    text: str
+    alarmTime: time
+    alarm: bool
+    alarmOneDayBefore: bool
+    color: str
+    textColor: str
+    sequentialCode: int
+    textSize: float
+    alarmTime2: time
+    alarm2: bool
+    alarmTwoDayBefore: bool
+    sound1: Optional[str]
+    sound2: Optional[str]
+    abbreviation: str
+    startTime1: time
+    endTime1: time
+    startTime2: time
+    endTime2: time
+    splitShift: bool
+    performAction: Optional[str]
+    hourlyEarnings: Optional[float]
+    extraHourlyEarnings: Optional[float]
+    restTime: Optional[float]
+    shiftTime: Optional[str]
+    calculatedManualShiftDuration: Optional[float]
+
+    @classmethod
+    def from_dict(cls, shift: dict[str, Any]):
+        # Turn boolean integers into actual python booleans
+        boolean_labels = ["alarm", "alarmOneDayBefore", "alarm2", "alarmTwoDayBefore", "splitShift"]
+        for label in boolean_labels:
+            if shift.get(label, 0) == 0:
+                shift[label] = False
+            else:
+                shift[label] = True
+
+        # Turn color 32 bit integers into hex color codes
+        color_labels = ["color", "textColor"]
+        for label in color_labels:
+            color_int = shift.get(label)
+            if color_int:
+                blue = color_int & 0xff
+                green = (color_int >> 8) & 0xff
+                red = (color_int >> 16) & 0xff
+
+                color_hex = ('#%02x%02x%02x' % (red, green, blue))
+                shift[label] = color_hex
+
+        # Turn time string into datetime.time object
+        time_labels = ["alarmTime", "alarmTime2", "startTime1", "endTime1", "startTime2", "endTime2"]
+        for label in time_labels:
+            time_value = shift.get(label)
+            if time_value:
+                datetime_object = datetime.strptime(time_value, '%H:%M').time()
+                shift[label] = datetime_object
+        return cls(**shift)
+
+
+@dataclass
+class DayWithShift(Shift):
+    date: date
+    note: str | None
+
+
+@dataclass
+class Day():
+    date: date
+    shift_id: Optional[int]
+    note: Optional[str]
+
+
+def validate_shifter_file(db: Connection):
+    table_names: list[tuple[str]] = fetch_tables(db)
+
+    # Check if table format is correct
+    table_name_list = [row[0] for row in table_names]
+    if ('tablaTurnos' not in table_name_list) or ('dias' not in table_name_list):
+        return False
+    else:
+        return True
+
+
+def fetch_tables(database: Connection) -> list[tuple[str]]:
+    """Find all tables in database and return them as a list"""
+    query = "SELECT name FROM sqlite_schema WHERE type = 'table'"
+    tables = database.execute(query).fetchall()
+    return tables
+
+
+def fetch_shift_types(database: Connection) -> dict[int, Shift]:
+    """Fetch different types of shift and format them as a dict of _id and Shift object"""
+
+    # Fetch all shift types from database
+    query = "SELECT * FROM tablaTurnos"
+    sql_shifts: list[tuple[Any]] = database.execute(query).fetchall()
+
+    # Pair every shift value with a label from shift_labels
+    dict_shifts: list = [dict(zip(shift_lables, shift)) for shift in sql_shifts]
+
+    # Turn every shift into a Shift object
+    shifts: dict[int, Shift] = {}
+    for shift in dict_shifts:
+        shift_object = Shift.from_dict(shift)
+        shifts[shift_object._id] = shift_object
+
+    return shifts
+
+
+def fetch_calendar(database: Connection) -> list[Any]:
+    """Fetch all notes from database and return them"""
+
+    # Fetch dates, shifts, and notes from table containing the dates
+    query = "SELECT fecha, turno1, notas FROM dias"
+    calendar: list[tuple[int, int | None, str | None]] = database.execute(query).fetchall()
+
+    formatted_calendar: list[Day] = []
+
+    for calendar_day in calendar:
+        # Turn SQL date into a datetime.date object
+        try:
+            insane_date_str = str(calendar_day[0])  # Format is YYYYMMDD, but the month is apperantly in integers, so january is 00, this has wasted 2 hours of my life (╯°□°)╯︵ ┻━┻
+            year = insane_date_str[:4]
+            month = int(insane_date_str[4:6]) + 1
+            month_str = f"{month:02d}"
+            day = insane_date_str[6:]
+
+            sane_date_str = f"{year}{month_str}{day}"
+
+            date_object = datetime.strptime(sane_date_str, "%Y%m%d").date()
+        except ValueError:
+            continue
+
+        # Get shift type and note of day
+        shift_type: int | None = calendar_day[1]
+        note: str | None = calendar_day[2]
+
+        # Unformated notes look like this: <p dir="ltr">TEXT_HERE</p>
+        # To parse this into normal text a regex match is used
+        if note:
+            note = re.findall(r'>(.*?)<', note)[0]
+
+        # Add formatted day to list as a Day object
+        formatted_calendar.append(Day(date_object, shift_type, note))
+
+    return formatted_calendar
+
+
+def combine_calendar_and_shifts(days: list[Day], shifts: dict[int, Shift]) -> list[Day | DayWithShift]:
+    """Combine calendar entries with details about shifts"""
+    combined_calendar: list[Day | DayWithShift] = []
+
+    for day in days:
+        shift_id = day.shift_id
+        if shift_id:
+            shift: Shift = shifts[shift_id]
+            shift_day = DayWithShift(date=day.date, note=day.note, **asdict(shift))
+            combined_calendar.append(shift_day)
+        else:
+            combined_calendar.append(day)
+
+    return combined_calendar
+
+
+def add_ics_event(calendar: ics.Calendar, name: str, start_date: date, start_time: time = time(0, 0), end_time=None, all_day=False, abreviation=None, second_part=False):
+    event = ics.Event()
+
+    event.name = name
+    event.begin = datetime.combine(start_date, start_time).replace(tzinfo=None).replace(tzinfo=local_tz)
+
+    if all_day:
+        event.make_all_day()
+        event.uid = f"note-{start_date.strftime('%Y-%m-%d')}-{name[:10].replace(' ', '')}"
+    elif end_time:
+        end_date = start_date if not (second_part or (start_time > end_time)) else start_date + timedelta(days=1)
+        event.end = datetime.combine(end_date, end_time).replace(tzinfo=local_tz)
+        event.uid = f"{abreviation if abreviation else 'shift'}-{start_date.strftime('%Y-%m-%d')}-{start_time.strftime('%H%M')}"
+    else:
+        print("ERROR: no end time found")
+
+    calendar.events.add(event)
+
+
+def shifts_to_ics(shifts: list[Day | DayWithShift], output_dir):
+    ics_calendar = ics.Calendar()
+
+    for day in shifts:
+        if isinstance(day, DayWithShift):
+            if day.startTime1 == day.endTime1:
+                add_ics_event(ics_calendar, day.text, day.date, abreviation=day.abbreviation, all_day=True)
+            else:
+                add_ics_event(ics_calendar, day.text, day.date, start_time=day.startTime1, end_time=day.endTime1, abreviation=day.abbreviation)
+                if day.splitShift:
+                    add_ics_event(ics_calendar, day.text, day.date, start_time=day.startTime2, end_time=day.endTime2, abreviation=f"{day.abbreviation}-2", second_part=True)
+        if day.note:
+            add_ics_event(ics_calendar, day.note, day.date, all_day=True)
+
+    with open(os.path.join(output_dir, 'output_calendar.ics'), 'w') as f:
+        f.write(ics_calendar.serialize())
+
+
+def main(db: Connection, output_dir: str):
+    # Get information of shift types
+    shift_types: dict[int, Shift] = fetch_shift_types(db)
+
+    # Get all days and their corresponding shift_type and notes
+    calendar: list[Day] = fetch_calendar(db)
+
+    # Add shift details to the calendar items
+    combined_calendar = combine_calendar_and_shifts(calendar, shift_types)
+
+    shifts_to_ics(combined_calendar, output_dir)
+
+    db.close()

shifter_converter-0.0.1/src/shifter_converter.egg-info/PKG-INFO

@@ -0,0 +1,19 @@
+Metadata-Version: 2.4
+Name: shifter-converter
+Version: 0.0.1
+Summary: A program for converting .Shifter files into .ics files
+Description-Content-Type: text/markdown
+Requires-Dist: ics>=0.7.3
+Requires-Dist: tzlocal>=5.3.1
+Requires-Dist: rich>=15.0.0
+
+# Shifter-Converter
+
+A program to convert .Shifter files to .ics files
+
+## AI
+
+AI was used for the following:
+
+- Formatting output of database commands
+- Instructions on setting up PyPI project

shifter_converter-0.0.1/src/shifter_converter.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+README.md
+pyproject.toml
+src/shifter_converter/__init__.py
+src/shifter_converter/cli.py
+src/shifter_converter/converter.py
+src/shifter_converter.egg-info/PKG-INFO
+src/shifter_converter.egg-info/SOURCES.txt
+src/shifter_converter.egg-info/dependency_links.txt
+src/shifter_converter.egg-info/entry_points.txt
+src/shifter_converter.egg-info/requires.txt
+src/shifter_converter.egg-info/top_level.txt

shifter_converter-0.0.1/src/shifter_converter.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

shifter_converter-0.0.1/src/shifter_converter.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+shifter-converter = shifter_converter.cli:main

shifter_converter-0.0.1/src/shifter_converter.egg-info/requires.txt

@@ -0,0 +1,3 @@
+ics>=0.7.3
+tzlocal>=5.3.1
+rich>=15.0.0

shifter_converter-0.0.1/src/shifter_converter.egg-info/top_level.txt

@@ -0,0 +1 @@
+shifter_converter