senaite.core.api.dtime.get_timezone() - Code Metrics - Inspection of "Ensure valid timezone in DX datetime field setter" - senaite/senaite.core - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Pull Request — 2.x (#1930)

by Ramon

created 2022-02-05 11:39 UTC

senaite.core.api.dtime.get_timezone() C

↳ Parent: senaite.core.api.dtime

Complexity

Conditions

Size

Total Lines	29
Code Lines	19

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
eloc	19
dl	0
loc	29
rs	6.6666
c	0
b	0
f	0
cc	9
nop	2

# -*- coding: utf-8 -*-

import os
import time
from datetime import date
from datetime import datetime

import six

import pytz
from bika.lims import logger
from bika.lims.api import APIError
from DateTime import DateTime
from DateTime.DateTime import DateError
from DateTime.DateTime import SyntaxError
from DateTime.DateTime import TimeError


def is_str(obj):
    """Check if the given object is a string

    :param obj: arbitrary object
    :returns: True when the object is a string
    """
    return isinstance(obj, six.string_types)


def is_d(dt):
    """Check if the date is a Python `date` object

    :param dt: date to check
    :returns: True when the date is a Python `date`
    """
    return type(dt) is date


def is_dt(dt):
    """Check if the date is a Python `datetime` object

    :param dt: date to check
    :returns: True when the date is a Python `datetime`
    """
    return type(dt) is datetime


def is_DT(dt):
    """Check if the date is a Zope `DateTime` object

    :param dt: object to check
    :returns: True when the object is a Zope `DateTime`
    """
    return type(dt) is DateTime


def is_date(dt):
    """Check if the date is a datetime or DateTime object

    :param dt: date to check
    :returns: True when the object is either a datetime or DateTime
    """
    if is_str(dt):
        DT = to_DT(dt)
        return is_date(DT)
    if is_d(dt):
        return True
    if is_dt(dt):
        return True
    if is_DT(dt):
        return True
    return False


def is_timezone_naive(dt):
    """Check if the date is timezone naive

    :param dt: date to check
    :returns: True when the date has no timezone
    """
    if is_d(dt):
        return True
    elif is_DT(dt):
        return dt.timezoneNaive()
    elif is_dt(dt):
        return dt.tzinfo is None
    elif is_str(dt):
        DT = to_DT(dt)
        return is_timezone_naive(DT)
    raise APIError("Expected a date type, got '%r'" % type(dt))


def is_timezone_aware(dt):
    """Check if the date is timezone aware

    :param dt: date to check
    :returns: True when the date has a timezone
    """
    return not is_timezone_naive(dt)


def to_DT(dt):
    """Convert to DateTime

    :param dt: DateTime/datetime/date
    :returns: DateTime object
    """
    if is_DT(dt):
        return dt
    elif is_str(dt):
        try:
            return DateTime(dt)
        except (DateError, TimeError, SyntaxError, IndexError):
            return None
    elif is_dt(dt):
        return DateTime(dt.isoformat())
    elif is_d(dt):
        dt = datetime(dt.year, dt.month, dt.day)
        return DateTime(dt.isoformat())
    else:
        return None


def to_dt(dt):
    """Convert to datetime

    :param dt: DateTime/datetime/date
    :returns: datetime object
    """
    if is_DT(dt):
        # get a valid pytz timezone
        tz = get_timezone(dt)
        dt = dt.asdatetime()
        if is_valid_timezone(tz):
            dt = to_zone(dt, tz)
        return dt
    elif is_str(dt):
        DT = to_DT(dt)
        return to_dt(DT)
    elif is_dt(dt):
        return dt
    elif is_d(dt):
        return datetime(dt.year, dt.month, dt.day)
    else:
        return None


def get_timezone(dt, default="UTC"):
    """Get a valid pytz timezone of the datetime object

    :param dt: date object
    :returns: timezone as string, e.g. Etc/GMT or CET
    """
    tz = None
    if is_dt(dt):
        tz = dt.tzname()
    elif is_DT(dt):
        tz = dt.timezone()
    elif is_d(dt):
        tz = default

    if tz:
        # convert DateTime `GMT` to `Etc/GMT` timezones
        # NOTE: `GMT+1` get `Etc/GMT-1`!
        if tz.startswith("GMT+0"):
            tz = tz.replace("GMT+0", "Etc/GMT")
        elif tz.startswith("GMT+"):
            tz = tz.replace("GMT+", "Etc/GMT-")
        elif tz.startswith("GMT-"):
            tz = tz.replace("GMT-", "Etc/GMT+")
        elif tz.startswith("GMT"):
            tz = tz.replace("GMT", "Etc/GMT")
    else:
        tz = default

    return tz


def is_valid_timezone(timezone):
    """Checks if the timezone is a valid pytz/Olson name

    :param timezone: pytz/Olson timezone name
    :returns: True when the timezone is a valid zone
    """
    try:
        pytz.timezone(timezone)
        return True
    except pytz.UnknownTimeZoneError:
        return False


def get_os_timezone():
    """Return the default timezone of the system

    :returns: OS timezone or UTC
    """
    fallback = "UTC"
    timezone = None
    if "TZ" in os.environ.keys():
        # Timezone from OS env var
        timezone = os.environ["TZ"]
    if not timezone:
        # Timezone from python time
        zones = time.tzname
        if zones and len(zones) > 0:
            timezone = zones[0]
        else:
            # Default fallback = UTC
            logger.warn(
                "Operating system\'s timezone cannot be found. "
                "Falling back to UTC.")
            timezone = fallback
    if not is_valid_timezone(timezone):
        return fallback
    return timezone


def to_zone(dt, timezone):
    """Convert date to timezone

    Adds the timezone for timezone naive datetimes

    :param dt: date object
    :param timezone: timezone
    :returns: date converted to timezone
    """
    if is_dt(dt) or is_d(dt):
        dt = to_dt(dt)
        zone = pytz.timezone(timezone)
        if is_timezone_aware(dt):
            return dt.astimezone(zone)
        return zone.localize(dt)
    elif is_DT(dt):
        # NOTE: This shifts the time according to the TZ offset
        return dt.toZone(timezone)
    raise TypeError("Expected a date, got '%r'" % type(dt))


def to_timestamp(dt):
    """Generate a Portable Operating System Interface (POSIX) timestamp

    :param dt: date object
    :returns: timestamp in seconds
    """
    timestamp = 0
    if is_DT(dt):
        timestamp = dt.timeTime()
    elif is_dt(dt):
        timestamp = time.mktime(dt.timetuple())
    elif is_str(dt):
        DT = to_DT(dt)
        return to_timestamp(DT)
    return timestamp


def from_timestamp(timestamp):
    """Generate a datetime object from a POSIX timestamp

    :param timestamp: POSIX timestamp
    :returns: datetime object
    """
    return datetime.utcfromtimestamp(timestamp)


def to_iso_format(dt):
    """Convert to ISO format
    """
    if is_dt(dt):
        return dt.isoformat()
    elif is_DT(dt):
        return dt.ISO()
    elif is_str(dt):
        DT = to_DT(dt)
        return to_iso_format(DT)
    return None


1			# -- coding: utf-8 --
2
3			import os
4			import time
5			from datetime import date
6			from datetime import datetime
7
8			import six
9
10			import pytz
11			from bika.lims import logger
12			from bika.lims.api import APIError
13			from DateTime import DateTime
14			from DateTime.DateTime import DateError
15			from DateTime.DateTime import SyntaxError
16			from DateTime.DateTime import TimeError
17
18
19			def is_str(obj):
20			"""Check if the given object is a string
21
22			:param obj: arbitrary object
23			:returns: True when the object is a string
24			"""
25			return isinstance(obj, six.string_types)
26
27
28			def is_d(dt):
29			"""Check if the date is a Python `date` object
30
31			:param dt: date to check
32			:returns: True when the date is a Python `date`
33			"""
34			return type(dt) is date
35
36
37			def is_dt(dt):
38			"""Check if the date is a Python `datetime` object
39
40			:param dt: date to check
41			:returns: True when the date is a Python `datetime`
42			"""
43			return type(dt) is datetime
44
45
46			def is_DT(dt):
47			"""Check if the date is a Zope `DateTime` object
48
49			:param dt: object to check
50			:returns: True when the object is a Zope `DateTime`
51			"""
52			return type(dt) is DateTime
53
54
55			def is_date(dt):
56			"""Check if the date is a datetime or DateTime object
57
58			:param dt: date to check
59			:returns: True when the object is either a datetime or DateTime
60			"""
61			if is_str(dt):
62			DT = to_DT(dt)
63			return is_date(DT)
64			if is_d(dt):
65			return True
66			if is_dt(dt):
67			return True
68			if is_DT(dt):
69			return True
70			return False
71
72
73			def is_timezone_naive(dt):
74			"""Check if the date is timezone naive
75
76			:param dt: date to check
77			:returns: True when the date has no timezone
78			"""
79			if is_d(dt):
80			return True
81			elif is_DT(dt):
82			return dt.timezoneNaive()
83			elif is_dt(dt):
84			return dt.tzinfo is None
85			elif is_str(dt):
86			DT = to_DT(dt)
87			return is_timezone_naive(DT)
88			raise APIError("Expected a date type, got '%r'" % type(dt))
89
90
91			def is_timezone_aware(dt):
92			"""Check if the date is timezone aware
93
94			:param dt: date to check
95			:returns: True when the date has a timezone
96			"""
97			return not is_timezone_naive(dt)
98
99
100			def to_DT(dt):
101			"""Convert to DateTime
102
103			:param dt: DateTime/datetime/date
104			:returns: DateTime object
105			"""
106			if is_DT(dt):
107			return dt
108			elif is_str(dt):
109			try:
110			return DateTime(dt)
111			except (DateError, TimeError, SyntaxError, IndexError):
112			return None
113			elif is_dt(dt):
114			return DateTime(dt.isoformat())
115			elif is_d(dt):
116			dt = datetime(dt.year, dt.month, dt.day)
117			return DateTime(dt.isoformat())
118			else:
119			return None
120
121
122			def to_dt(dt):
123			"""Convert to datetime
124
125			:param dt: DateTime/datetime/date
126			:returns: datetime object
127			"""
128			if is_DT(dt):
129			# get a valid pytz timezone
130			tz = get_timezone(dt)
131			dt = dt.asdatetime()
132			if is_valid_timezone(tz):
133			dt = to_zone(dt, tz)
134			return dt
135			elif is_str(dt):
136			DT = to_DT(dt)
137			return to_dt(DT)
138			elif is_dt(dt):
139			return dt
140			elif is_d(dt):
141			return datetime(dt.year, dt.month, dt.day)
142			else:
143			return None
144
145
146			def get_timezone(dt, default="UTC"):
147			"""Get a valid pytz timezone of the datetime object
148
149			:param dt: date object
150			:returns: timezone as string, e.g. Etc/GMT or CET
151			"""
152			tz = None
153			if is_dt(dt):
154			tz = dt.tzname()
155			elif is_DT(dt):
156			tz = dt.timezone()
157			elif is_d(dt):
158			tz = default
159
160			if tz:
161			# convert DateTime `GMT` to `Etc/GMT` timezones
162			# NOTE: `GMT+1` get `Etc/GMT-1`!
163			if tz.startswith("GMT+0"):
164			tz = tz.replace("GMT+0", "Etc/GMT")
165			elif tz.startswith("GMT+"):
166			tz = tz.replace("GMT+", "Etc/GMT-")
167			elif tz.startswith("GMT-"):
168			tz = tz.replace("GMT-", "Etc/GMT+")
169			elif tz.startswith("GMT"):
170			tz = tz.replace("GMT", "Etc/GMT")
171			else:
172			tz = default
173
174			return tz
175
176
177			def is_valid_timezone(timezone):
178			"""Checks if the timezone is a valid pytz/Olson name
179
180			:param timezone: pytz/Olson timezone name
181			:returns: True when the timezone is a valid zone
182			"""
183			try:
184			pytz.timezone(timezone)
185			return True
186			except pytz.UnknownTimeZoneError:
187			return False
188
189
190			def get_os_timezone():
191			"""Return the default timezone of the system
192
193			:returns: OS timezone or UTC
194			"""
195			fallback = "UTC"
196			timezone = None
197			if "TZ" in os.environ.keys():
198			# Timezone from OS env var
199			timezone = os.environ["TZ"]
200			if not timezone:
201			# Timezone from python time
202			zones = time.tzname
203			if zones and len(zones) > 0:
204			timezone = zones[0]
205			else:
206			# Default fallback = UTC
207			logger.warn(
208			"Operating system\'s timezone cannot be found. "
209			"Falling back to UTC.")
210			timezone = fallback
211			if not is_valid_timezone(timezone):
212			return fallback
213			return timezone
214
215
216			def to_zone(dt, timezone):
217			"""Convert date to timezone
218
219			Adds the timezone for timezone naive datetimes
220
221			:param dt: date object
222			:param timezone: timezone
223			:returns: date converted to timezone
224			"""
225			if is_dt(dt) or is_d(dt):
226			dt = to_dt(dt)
227			zone = pytz.timezone(timezone)
228			if is_timezone_aware(dt):
229			return dt.astimezone(zone)
230			return zone.localize(dt)
231			elif is_DT(dt):
232			# NOTE: This shifts the time according to the TZ offset
233			return dt.toZone(timezone)
234			raise TypeError("Expected a date, got '%r'" % type(dt))
235
236
237			def to_timestamp(dt):
238			"""Generate a Portable Operating System Interface (POSIX) timestamp
239
240			:param dt: date object
241			:returns: timestamp in seconds
242			"""
243			timestamp = 0
244			if is_DT(dt):
245			timestamp = dt.timeTime()
246			elif is_dt(dt):
247			timestamp = time.mktime(dt.timetuple())
248			elif is_str(dt):
249			DT = to_DT(dt)
250			return to_timestamp(DT)
251			return timestamp
252
253
254			def from_timestamp(timestamp):
255			"""Generate a datetime object from a POSIX timestamp
256
257			:param timestamp: POSIX timestamp
258			:returns: datetime object
259			"""
260			return datetime.utcfromtimestamp(timestamp)
261
262
263			def to_iso_format(dt):
264			"""Convert to ISO format
265			"""
266			if is_dt(dt):
267			return dt.isoformat()
268			elif is_DT(dt):
269			return dt.ISO()
270			elif is_str(dt):
271			DT = to_DT(dt)
272			return to_iso_format(DT)
273			return None
274

senaite / senaite.core

Pull Request — 2.x (#1930)

senaite.core.api.dtime.get_timezone() C

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like