##// END OF EJS Templates
Format existing code with black
Nils Müller -
Show More
@@ -5,25 +5,26 b' Timezone utilities'
5 Just UTC-awareness right now
5 Just UTC-awareness right now
6 """
6 """
7
7
8 #-----------------------------------------------------------------------------
8 # -----------------------------------------------------------------------------
9 # Copyright (C) 2013 The IPython Development Team
9 # Copyright (C) 2013 The IPython Development Team
10 #
10 #
11 # Distributed under the terms of the BSD License. The full license is in
11 # Distributed under the terms of the BSD License. The full license is in
12 # the file COPYING, distributed as part of this software.
12 # the file COPYING, distributed as part of this software.
13 #-----------------------------------------------------------------------------
13 # -----------------------------------------------------------------------------
14
14
15 #-----------------------------------------------------------------------------
15 # -----------------------------------------------------------------------------
16 # Imports
16 # Imports
17 #-----------------------------------------------------------------------------
17 # -----------------------------------------------------------------------------
18
18
19 from datetime import tzinfo, timedelta, datetime
19 from datetime import tzinfo, timedelta, datetime
20
20
21 #-----------------------------------------------------------------------------
21 # -----------------------------------------------------------------------------
22 # Code
22 # Code
23 #-----------------------------------------------------------------------------
23 # -----------------------------------------------------------------------------
24 # constant for zero offset
24 # constant for zero offset
25 ZERO = timedelta(0)
25 ZERO = timedelta(0)
26
26
27
27 class tzUTC(tzinfo):
28 class tzUTC(tzinfo):
28 """tzinfo object for UTC (zero offset)"""
29 """tzinfo object for UTC (zero offset)"""
29
30
@@ -39,10 +40,13 b' UTC = tzUTC() # type: ignore[abstract]'
39
40
40 def utc_aware(unaware):
41 def utc_aware(unaware):
41 """decorator for adding UTC tzinfo to datetime's utcfoo methods"""
42 """decorator for adding UTC tzinfo to datetime's utcfoo methods"""
43
42 def utc_method(*args, **kwargs):
44 def utc_method(*args, **kwargs):
43 dt = unaware(*args, **kwargs)
45 dt = unaware(*args, **kwargs)
44 return dt.replace(tzinfo=UTC)
46 return dt.replace(tzinfo=UTC)
47
45 return utc_method
48 return utc_method
46
49
50
47 utcfromtimestamp = utc_aware(datetime.utcfromtimestamp)
51 utcfromtimestamp = utc_aware(datetime.utcfromtimestamp)
48 utcnow = utc_aware(datetime.utcnow)
52 utcnow = utc_aware(datetime.utcnow)
General Comments 0
You need to be logged in to leave comments. Login now