mirror of
https://github.com/moparisthebest/SickRage
synced 2024-11-16 14:25:02 -05:00
110 lines
3.5 KiB
Python
110 lines
3.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
Copyright (c) 2003-2005 Gustavo Niemeyer <gustavo@niemeyer.net>
|
|
|
|
This module offers extensions to the standard Python
|
|
datetime module.
|
|
"""
|
|
import logging
|
|
import os
|
|
from subprocess import call
|
|
from tarfile import TarFile
|
|
|
|
from dateutil.tz import tzfile
|
|
|
|
__author__ = "Tomi Pieviläinen <tomi.pievilainen@iki.fi>"
|
|
__license__ = "Simplified BSD"
|
|
|
|
__all__ = ["setcachesize", "gettz", "rebuild"]
|
|
|
|
CACHE = []
|
|
CACHESIZE = 10
|
|
|
|
class tzfile(tzfile):
|
|
def __reduce__(self):
|
|
return (gettz, (self._filename,))
|
|
|
|
def getzoneinfofile():
|
|
filenames = sorted(os.listdir(os.path.join(os.path.dirname(__file__))))
|
|
filenames.reverse()
|
|
for entry in filenames:
|
|
if entry.startswith("zoneinfo") and ".tar." in entry:
|
|
return os.path.join(os.path.dirname(__file__), entry)
|
|
return None
|
|
|
|
ZONEINFOFILE = getzoneinfofile()
|
|
|
|
del getzoneinfofile
|
|
|
|
def setcachesize(size):
|
|
global CACHESIZE, CACHE
|
|
CACHESIZE = size
|
|
del CACHE[size:]
|
|
|
|
def gettz(name):
|
|
tzinfo = None
|
|
if ZONEINFOFILE:
|
|
for cachedname, tzinfo in CACHE:
|
|
if cachedname == name:
|
|
break
|
|
else:
|
|
tf = TarFile.open(ZONEINFOFILE)
|
|
try:
|
|
zonefile = tf.extractfile(name)
|
|
except KeyError:
|
|
tzinfo = None
|
|
else:
|
|
tzinfo = tzfile(zonefile)
|
|
tf.close()
|
|
CACHE.insert(0, (name, tzinfo))
|
|
del CACHE[CACHESIZE:]
|
|
return tzinfo
|
|
|
|
def rebuild(filename, tag=None, format="gz"):
|
|
"""Rebuild the internal timezone info in dateutil/zoneinfo/zoneinfo*tar*
|
|
|
|
filename is the timezone tarball from ftp.iana.org/tz.
|
|
|
|
"""
|
|
import tempfile, shutil
|
|
tmpdir = tempfile.mkdtemp()
|
|
zonedir = os.path.join(tmpdir, "zoneinfo")
|
|
moduledir = os.path.dirname(__file__)
|
|
if tag: tag = "-"+tag
|
|
targetname = "zoneinfo%s.tar.%s" % (tag, format)
|
|
try:
|
|
tf = TarFile.open(filename)
|
|
# The "backwards" zone file contains links to other files, so must be
|
|
# processed as last
|
|
for name in sorted(tf.getnames(),
|
|
key=lambda k: k != "backward" and k or "z"):
|
|
if not (name.endswith(".sh") or
|
|
name.endswith(".tab") or
|
|
name == "leapseconds"):
|
|
tf.extract(name, tmpdir)
|
|
filepath = os.path.join(tmpdir, name)
|
|
try:
|
|
# zic will return errors for nontz files in the package
|
|
# such as the Makefile or README, so check_call cannot
|
|
# be used (or at least extra checks would be needed)
|
|
call(["zic", "-d", zonedir, filepath])
|
|
except OSError as e:
|
|
if e.errno == 2:
|
|
logging.error(
|
|
"Could not find zic. Perhaps you need to install "
|
|
"libc-bin or some other package that provides it, "
|
|
"or it's not in your PATH?")
|
|
raise
|
|
tf.close()
|
|
target = os.path.join(moduledir, targetname)
|
|
for entry in os.listdir(moduledir):
|
|
if entry.startswith("zoneinfo") and ".tar." in entry:
|
|
os.unlink(os.path.join(moduledir, entry))
|
|
tf = TarFile.open(target, "w:%s" % format)
|
|
for entry in os.listdir(zonedir):
|
|
entrypath = os.path.join(zonedir, entry)
|
|
tf.add(entrypath, entry)
|
|
tf.close()
|
|
finally:
|
|
shutil.rmtree(tmpdir)
|