feat/timezone_config

lingua_franca/config.py

@@ -1 +1,2 @@
 load_langs_on_demand = False
+inject_timezones = True

lingua_franca/internal.py

@@ -2,10 +2,12 @@
 from functools import wraps
 from importlib import import_module
 from inspect import signature
-from sys import version
-from warnings import warn
 
+from warnings import warn
+from datetime import datetime
 from lingua_franca import config
+from lingua_franca.time import to_local
+
 
 _SUPPORTED_LANGUAGES = ("ca", "cs", "da", "de", "en", "es", "fr", "hu",
                         "it", "nl", "pl", "pt", "sl", "sv")
@@ -456,10 +458,19 @@ def _call_localized_function(func, *args, **kwargs):
             lang_param_index = func_params.index('lang')
             full_lang_code = None
 
+            # Check if we need to add timezone awareness to any datetime object
+            if config.inject_timezones:
+                for k, v in kwargs.items():
+                    if isinstance(v, datetime):
+                        kwargs[k] = to_local(v)
+                for idx, v in enumerate(args):
+                    if isinstance(v, datetime) and v.tzinfo is None:
+                        args = args[:idx] + (to_local(v),) + args[idx + 1:]
+
             # Check if we're passing a lang as a kwarg
             if 'lang' in kwargs.keys():
                 lang_param = kwargs['lang']
-                if lang_param == None:
+                if lang_param is None:
                     warn(NoneLangWarning)
                     lang_code = get_default_lang()
                 else:
@@ -468,7 +479,7 @@ def _call_localized_function(func, *args, **kwargs):
             # Check if we're passing a lang as a positional arg
             elif lang_param_index < len(args):
                 lang_param = args[lang_param_index]
-                if lang_param == None:
+                if lang_param is None:
                     warn(NoneLangWarning)
                     lang_code = get_default_lang()
                 elif lang_param in _SUPPORTED_LANGUAGES or \

lingua_franca/lang/parse_ca.py

@@ -19,9 +19,10 @@
     TODO: numbers greater than 999999
     TODO: date time ca
 """
-
+from dateutil.tz import gettz
 from datetime import datetime
 from dateutil.relativedelta import relativedelta
+from lingua_franca.time import now_local
 from lingua_franca.lang.parse_common import is_numeric, look_for_fractions
 from lingua_franca.lang.common_data_ca import _NUMBERS_CA, \
     _FEMALE_DETERMINANTS_CA, _FEMALE_ENDINGS_CA, \
@@ -337,9 +338,10 @@ def date_found():
                 minAbs or secOffset != 0
             )
 
-    if text == "" or not anchorDate:
+    if text == "":
         return None
 
+    anchorDate = anchorDate or now_local()
     found = False
     daySpecified = False
     dayOffset = False
@@ -995,6 +997,9 @@ def date_found():
             datestr = datestr.replace(monthsShort[idx], en_month)
 
         temp = datetime.strptime(datestr, "%B %d")
+        if extractedDate.tzinfo:
+            temp = temp.replace(tzinfo=gettz("UTC"))
+            temp = temp.astimezone(extractedDate.tzinfo)
         if not hasYear:
             temp = temp.replace(year=extractedDate.year)
             if extractedDate < temp:

lingua_franca/lang/parse_cs.py

@@ -760,9 +760,10 @@ def date_found():
                 minAbs or secOffset != 0
             )
 
-    if text == "" or not anchorDate:
+    if text == "":
         return None
 
+    anchorDate = anchorDate or now_local()
     found = False
     daySpecified = False
     dayOffset = False

lingua_franca/lang/parse_da.py

@@ -13,12 +13,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
+from dateutil.tz import gettz
 from datetime import datetime
 from dateutil.relativedelta import relativedelta
 from lingua_franca.lang.parse_common import is_numeric, look_for_fractions, \
     extract_numbers_generic, Normalizer
 from lingua_franca.lang.common_data_da import _DA_NUMBERS
 from lingua_franca.lang.format_da import pronounce_number_da
+from lingua_franca.time import now_local
 
 
 def extract_number_da(text, short_scale=True, ordinals=False):
@@ -141,9 +143,10 @@ def date_found():
                 minAbs or secOffset != 0
             )
 
-    if text == "" or not anchorDate:
+    if text == "":
         return None
 
+    anchorDate = anchorDate or now_local()
     found = False
     daySpecified = False
     dayOffset = False
@@ -707,6 +710,10 @@ def date_found():
             datestr = datestr.replace(monthsShort[idx], en_month)
 
         temp = datetime.strptime(datestr, "%B %d")
+        if extractedDate.tzinfo:
+            temp = temp.replace(tzinfo=gettz("UTC"))
+            temp = temp.astimezone(extractedDate.tzinfo)
+
         if not hasYear:
             temp = temp.replace(year=extractedDate.year)
             if extractedDate < temp:

lingua_franca/lang/parse_de.py

@@ -14,12 +14,15 @@
 # limitations under the License.
 #
 import re
+from dateutil.tz import gettz
 from datetime import datetime, timedelta
 from dateutil.relativedelta import relativedelta
 from lingua_franca.lang.parse_common import is_numeric, look_for_fractions, \
     extract_numbers_generic, Normalizer
 from lingua_franca.lang.common_data_de import _DE_NUMBERS
 from lingua_franca.lang.format_de import pronounce_number_de
+from lingua_franca.time import now_local
+
 
 de_numbers = {
     'null': 0,
@@ -260,9 +263,10 @@ def date_found():
                 minAbs or secOffset != 0
             )
 
-    if text == "" or not anchorDate:
+    if text == "":
         return None
 
+    anchorDate = anchorDate or now_local()
     found = False
     daySpecified = False
     dayOffset = False
@@ -833,6 +837,10 @@ def date_found():
             datestr = datestr.replace(monthsShort[idx], en_month)
 
         temp = datetime.strptime(datestr, "%B %d")
+        if extractedDate.tzinfo:
+            temp = temp.replace(tzinfo=gettz("UTC"))
+            temp = temp.astimezone(extractedDate.tzinfo)
+
         if not hasYear:
             temp = temp.replace(year=extractedDate.year)
             if extractedDate < temp:

lingua_franca/lang/parse_en.py

@@ -17,6 +17,7 @@
 
 from dateutil.relativedelta import relativedelta
 
+from lingua_franca.time import now_local
 from lingua_franca.lang.parse_common import is_numeric, look_for_fractions, \
     invert_dict, ReplaceableNumber, partition_list, tokenize, Token, Normalizer
 from lingua_franca.lang.common_data_en import _ARTICLES_EN, _NUM_STRING_EN, \
@@ -671,8 +672,10 @@ def date_found():
                 hrAbs or minOffset != 0 or
                 minAbs or secOffset != 0
             )
+
     if not anchorDate:
-        anchorDate = datetime.now()
+        anchorDate = now_local()
+
     if text == "":
         return None
 

lingua_franca/lang/parse_es.py

@@ -14,8 +14,10 @@
 # limitations under the License.
 #
 from datetime import datetime
-from dateutil.relativedelta import relativedelta
 from dateutil.tz import gettz
+from dateutil.relativedelta import relativedelta
+
+from lingua_franca.time import now_local
 from lingua_franca.lang.format_es import pronounce_number_es
 from lingua_franca.lang.parse_common import *
 from lingua_franca.lang.common_data_es import _ARTICLES_ES, _STRING_NUM_ES
@@ -367,7 +369,7 @@ def date_found():
     if text == "":
         return None
     if anchorDate is None:
-        anchorDate = datetime.now()
+        anchorDate = now_local()
 
     found = False
     daySpecified = False
@@ -1020,17 +1022,19 @@ def date_found():
             datestr = datestr.replace(monthsShort[idx], en_month)
 
         temp = datetime.strptime(datestr, "%B %d")
-        temp = temp.replace(tzinfo=None)
+        if extractedDate.tzinfo:
+            temp = temp.replace(tzinfo=gettz("UTC"))
+            temp = temp.astimezone(extractedDate.tzinfo)
+        # temp = to_local(temp)
+
         if not hasYear:
             temp = temp.replace(year=extractedDate.year)
 
             if extractedDate < temp:
-                extractedDate = extractedDate.replace(year=int(currentYear),
-                                                      month=int(
-                                                          temp.strftime(
-                                                              "%m")),
-                                                      day=int(temp.strftime(
-                                                          "%d")))
+                extractedDate = extractedDate.replace(
+                    year=int(currentYear),
+                    month=int(temp.strftime("%m")),
+                    day=int(temp.strftime("%d")))
             else:
                 extractedDate = extractedDate.replace(
                     year=int(currentYear) + 1,

lingua_franca/lang/parse_fr.py

@@ -14,13 +14,16 @@
 # limitations under the License.
 #
 import re
+from dateutil.tz import gettz
 from datetime import datetime, timedelta
 from dateutil.relativedelta import relativedelta
 from lingua_franca.lang.parse_common import is_numeric, look_for_fractions, \
     extract_numbers_generic, Normalizer
 from lingua_franca.lang.format_fr import pronounce_number_fr
 from lingua_franca.lang.common_data_fr import _ARTICLES_FR, _NUMBERS_FR, \
     _ORDINAL_ENDINGS_FR
+from lingua_franca.time import now_local
+
 
 def extract_duration_fr(text):
     """
@@ -491,9 +494,10 @@ def date_found():
                 hrOffset != 0 or minOffset != 0 or secOffset != 0
             )
 
-    if text == "" or not anchorDate:
+    if text == "":
         return None
 
+    anchorDate = anchorDate or now_local()
     found = False
     daySpecified = False
     dayOffset = False
@@ -939,6 +943,9 @@ def date_found():
     if datestr != "":
         if not hasYear:
             temp = datetime.strptime(datestr, "%B %d")
+            if extractedDate.tzinfo:
+                temp = temp.replace(tzinfo=gettz("UTC"))
+                temp = temp.astimezone(extractedDate.tzinfo)
             temp = temp.replace(year=extractedDate.year)
             if extractedDate < temp:
                 extractedDate = extractedDate.replace(year=int(currentYear),

lingua_franca/lang/parse_hu.py

@@ -13,7 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-from datetime import datetime, timedelta
+from lingua_franca.time import now_local
 from lingua_franca.lang.parse_common import Normalizer
 
 

lingua_franca/lang/parse_it.py

@@ -21,6 +21,7 @@
 import collections
 from datetime import datetime
 from dateutil.relativedelta import relativedelta
+from lingua_franca.time import now_local
 from lingua_franca.lang.parse_common import is_numeric, look_for_fractions, \
     extract_numbers_generic, Normalizer
 from lingua_franca.lang.format_it import _LONG_SCALE_IT, _SHORT_SCALE_IT, \
@@ -493,9 +494,9 @@ def date_found():
                 month_offset != 0 or day_offset is True or hr_offset != 0 or
                 hr_abs or min_offset != 0 or min_abs or sec_offset != 0)
 
-    if text == '' or not anchorDate:
+    if text == '':
         return None
-
+    anchorDate = anchorDate or now_local()
     found = False
     day_specified = False
     day_offset = False

lingua_franca/lang/parse_nl.py

@@ -24,6 +24,7 @@
     _LONG_SCALE_NL, _MULTIPLIES_LONG_SCALE_NL, _MULTIPLIES_SHORT_SCALE_NL,\
     _NEGATIVES_NL, _SHORT_SCALE_NL, _STRING_LONG_ORDINAL_NL, _STRING_NUM_NL, \
     _STRING_SHORT_ORDINAL_NL, _SUMS_NL
+from lingua_franca.time import now_local
 import re
 
 
@@ -560,9 +561,10 @@ def date_found():
                 minAbs or secOffset != 0
             )
 
-    if text == "" or not anchorDate:
+    if text == "":
         return None
 
+    anchorDate = anchorDate or now_local()
     found = False
     daySpecified = False
     dayOffset = False

lingua_franca/lang/parse_pl.py

@@ -23,7 +23,7 @@
     _SHORT_SCALE_PL, _SHORT_ORDINAL_PL, _FRACTION_STRING_PL, _TIME_UNITS_CONVERSION, \
     _TIME_UNITS_NORMALIZATION, _MONTHS_TO_EN, _DAYS_TO_EN, _ORDINAL_BASE_PL, \
     _ALT_ORDINALS_PL
-
+from lingua_franca.time import now_local
 import re
 
 
@@ -710,9 +710,10 @@ def date_found():
                 minAbs or secOffset != 0
             )
 
-    if string == "" or not dateNow:
+    if string == "":
         return None
 
+    dateNow = dateNow or now_local()
     found = False
     daySpecified = False
     dayOffset = False

lingua_franca/lang/parse_pt.py

@@ -21,13 +21,15 @@
 """
 
 from datetime import datetime
+from dateutil.tz import gettz
 from dateutil.relativedelta import relativedelta
 from lingua_franca.lang.parse_common import is_numeric, look_for_fractions
 from lingua_franca.lang.common_data_pt import _NUMBERS_PT, \
     _FEMALE_DETERMINANTS_PT, _FEMALE_ENDINGS_PT, \
     _MALE_DETERMINANTS_PT, _MALE_ENDINGS_PT, _GENDERS_PT
 from lingua_franca.internal import resolve_resource_file
 from lingua_franca.lang.parse_common import Normalizer
+from lingua_franca.time import now_local
 import json
 import re
 
@@ -286,9 +288,10 @@ def date_found():
                 minAbs or secOffset != 0
             )
 
-    if text == "" or not anchorDate:
+    if text == "":
         return None
 
+    anchorDate = anchorDate or now_local()
     found = False
     daySpecified = False
     dayOffset = False
@@ -953,6 +956,10 @@ def date_found():
             datestr = datestr.replace(monthsShort[idx], en_month)
 
         temp = datetime.strptime(datestr, "%B %d")
+        if extractedDate.tzinfo:
+            temp = temp.replace(tzinfo=gettz("UTC"))
+            temp = temp.astimezone(extractedDate.tzinfo)
+
         if not hasYear:
             temp = temp.replace(year=extractedDate.year)
             if extractedDate < temp: