From 443d9bc028c1969823482325222ad71374200073 Mon Sep 17 00:00:00 2001
From: Jonathan Weth <git@jonathanweth.de>
Date: Sat, 28 Mar 2020 17:05:09 +0100
Subject: [PATCH] Add typing to helper functions for UNTIS dates
---
aleksis/apps/untis/util/mysql/api_helper.py | 14 ++++++++------
1 file changed, 8 insertions(+), 6 deletions(-)
diff --git a/aleksis/apps/untis/util/mysql/api_helper.py b/aleksis/apps/untis/util/mysql/api_helper.py
index 0cedd67..55c95f8 100644
--- a/aleksis/apps/untis/util/mysql/api_helper.py
+++ b/aleksis/apps/untis/util/mysql/api_helper.py
@@ -1,4 +1,4 @@
-from django.utils import timezone
+from datetime import date, datetime
def clean_array(a, conv=None):
@@ -23,15 +23,17 @@ def untis_split_third(s, conv=None):
return clean_array(s.split(";"), conv=conv)
-DATE_FORMAT = "%Y%m%d"
+UNTIS_DATE_FORMAT = "%Y%m%d"
-def untis_date_to_date(untis):
- return timezone.datetime.strptime(str(untis), DATE_FORMAT)
+def untis_date_to_date(untis: int) -> date:
+ """ Converts a UNTIS date to a python date """
+ return datetime.strptime(str(untis), UNTIS_DATE_FORMAT).date()
-def date_to_untis_date(date):
- return date.strftime(DATE_FORMAT)
+def date_to_untis_date(date: date) -> int:
+ """ Converts a python date to a UNTIS date """
+ return int(date.strftime(UNTIS_DATE_FORMAT))
def untis_colour_to_hex(colour: int) -> str:
--
GitLab