mirror of
https://github.com/wassname/catalyst.git
synced 2026-07-08 07:12:55 +08:00
Stephen Diehl
52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
"""
|
|
Small classes to assist with timezone calculations, LOGGER configuration,
|
|
and other common operations.
|
|
"""
|
|
|
|
import datetime
|
|
import pytz
|
|
import logging
|
|
|
|
LOGGER = logging.getLogger('QSimLogger')
|
|
|
|
def configure_logging(loglevel=logging.DEBUG):
|
|
"""
|
|
Configures zipline.util.LOGGER to write a rotating file
|
|
(10M per file, 5 files) to `` /var/log/zipline.log ``.
|
|
"""
|
|
LOGGER.setLevel(loglevel)
|
|
handler = logging.handlers.RotatingFileHandler(
|
|
"/var/log/zipline/{lfn}.log".format(lfn="zipline"),
|
|
maxBytes=10*1024*1024, backupCount=5
|
|
)
|
|
handler.setFormatter(logging.Formatter(
|
|
"%(asctime)s %(levelname)s %(filename)s %(funcName)s - %(message)s",
|
|
"%Y-%m-%d %H:%M:%S %Z")
|
|
)
|
|
LOGGER.addHandler(handler)
|
|
LOGGER.info("logging started...")
|
|
|
|
def parse_date(dt_str):
|
|
"""
|
|
Parse strings according to the same format as generated by
|
|
format_date.
|
|
"""
|
|
if(dt_str == None):
|
|
return None
|
|
parts = dt_str.split(".")
|
|
dt = datetime.datetime.strptime(parts[0], '%Y/%m/%d-%H:%M:%S').replace(
|
|
microsecond=int(parts[1]+"000")).replace(tzinfo = pytz.utc
|
|
)
|
|
return dt
|
|
|
|
def format_date(dt):
|
|
"""
|
|
Format the date into a date with millesecond resolution and
|
|
string/alphabetical sorting that is equivalent to datetime sorting.
|
|
"""
|
|
if(dt == None):
|
|
return None
|
|
dt_str = dt.strftime('%Y/%m/%d-%H:%M:%S') + "." + str(dt.microsecond / 1000)
|
|
return dt_str
|
|
|