SickGear/sickbeard/sbdatetime.py

# Author: Nic Wolfe <nic@wolfeden.ca>
# URL: http://code.google.com/p/sickbeard/
#
# This file is part of SickGear.
#
# SickGear is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# SickGear is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with SickGear.  If not, see <http://www.gnu.org/licenses/>.

import datetime
import locale
import functools
import re
import time

import sickbeard
from sickbeard.network_timezones import sb_timezone

date_presets = ('%Y-%m-%d',
                '%a, %Y-%m-%d',
                '%A, %Y-%m-%d',
                '%y-%m-%d',
                '%a, %y-%m-%d',
                '%A, %y-%m-%d',
                '%m/%d/%Y',
                '%a, %m/%d/%Y',
                '%A, %m/%d/%Y',
                '%m/%d/%y',
                '%a, %m/%d/%y',
                '%A, %m/%d/%y',
                '%m-%d-%Y',
                '%a, %m-%d-%Y',
                '%A, %m-%d-%Y',
                '%m-%d-%y',
                '%a, %m-%d-%y',
                '%A, %m-%d-%y',
                '%m.%d.%Y',
                '%a, %m.%d.%Y',
                '%A, %m.%d.%Y',
                '%m.%d.%y',
                '%a, %m.%d.%y',
                '%A, %m.%d.%y',
                '%d-%m-%Y',
                '%a, %d-%m-%Y',
                '%A, %d-%m-%Y',
                '%d-%m-%y',
                '%a, %d-%m-%y',
                '%A, %d-%m-%y',
                '%d/%m/%Y',
                '%a, %d/%m/%Y',
                '%A, %d/%m/%Y',
                '%d/%m/%y',
                '%a, %d/%m/%y',
                '%A, %d/%m/%y',
                '%d.%m.%Y',
                '%a, %d.%m.%Y',
                '%A, %d.%m.%Y',
                '%d.%m.%y',
                '%a, %d.%m.%y',
                '%A, %d.%m.%y',
                '%d. %b %Y',
                '%a, %d. %b %Y',
                '%A, %d. %b %Y',
                '%d. %b %y',
                '%a, %d. %b %y',
                '%A, %d. %b %y',
                '%d. %B %Y',
                '%a, %d. %B %Y',
                '%A, %d. %B %Y',
                '%d. %B %y',
                '%a, %d. %B %y',
                '%A, %d. %B %y',
                '%b %d, %Y',
                '%a, %b %d, %Y',
                '%A, %b %d, %Y',
                '%B %d, %Y',
                '%a, %B %d, %Y',
                '%A, %B %d, %Y')

time_presets = ('%I:%M:%S %p',
                '%I:%M:%S %P',
                '%H:%M:%S')


# helper class
class static_or_instance(object):
    def __init__(self, func):
        self.func = func

    def __get__(self, instance, owner):
        return functools.partial(self.func, instance)


# subclass datetime.datetime to add function to display custom date and time formats
class sbdatetime(datetime.datetime):
    has_locale = True

    @static_or_instance
    def convert_to_setting(self, dt=None):
        obj = (dt, self)[self is not None]
        try:
            if 'local' == sickbeard.TIMEZONE_DISPLAY:
                return obj.astimezone(sb_timezone)
        except:
            pass

        return obj

    @static_or_instance
    def setlocale(self, setlocale=True, use_has_locale=None, locale_str=''):
        if setlocale:
            try:
                if None is use_has_locale or use_has_locale:
                    locale.setlocale(locale.LC_TIME, locale_str)
            except:
                if None is not use_has_locale:
                    sbdatetime.has_locale = False
                pass

    # display Time in SickGear Format
    @static_or_instance
    def sbftime(self, dt=None, show_seconds=False, t_preset=None, setlocale=True, markup=False):

        sbdatetime.setlocale(setlocale=setlocale, use_has_locale=sbdatetime.has_locale, locale_str='us_US')

        strt = ''
        try:
            obj = (dt, self)[self is not None]
            if None is not obj:
                tmpl = (((sickbeard.TIME_PRESET, sickbeard.TIME_PRESET_W_SECONDS)[show_seconds]),
                        t_preset)[None is not t_preset]
                tmpl = (tmpl.replace(':%S', ''), tmpl)[show_seconds]

                strt = obj.strftime(tmpl.replace('%P', '%p'))

                if sickbeard.TRIM_ZERO:
                    strt = re.sub(r'^0(\d:\d\d)', r'\1', strt)

                if re.search(r'(?im)%p$', tmpl):
                    if '%p' in tmpl:
                        strt = strt.upper()
                    elif '%P' in tmpl:
                        strt = strt.lower()

                    if sickbeard.TRIM_ZERO:
                        strt = re.sub(r'(?im)^(\d+)(?::00)?(\s?[ap]m)', r'\1\2', strt)

                if markup:
                    match = re.search(r'(?im)(\d{1,2})(?:(.)(\d\d)(?:(.)(\d\d))?)?(?:\s?([ap]m))?$', strt)
                    if match:
                        strt = ('%s%s%s%s%s%s' % (
                            ('<span class="time-hr">%s</span>' % match.group(1), '')[None is match.group(1)],
                            ('<span class="time-hr-min">%s</span>' % match.group(2), '')[None is match.group(2)],
                            ('<span class="time-min">%s</span>' % match.group(3), '')[None is match.group(3)],
                            ('<span class="time-min-sec">%s</span>' % match.group(4), '')[None is match.group(4)],
                            ('<span class="time-sec">%s</span>' % match.group(5), '')[None is match.group(5)],
                            ('<span class="time-am-pm">%s</span>' % match.group(6), '')[None is match.group(6)]))

        finally:
            sbdatetime.setlocale(setlocale=setlocale, use_has_locale=sbdatetime.has_locale)
            return strt

    # display Date in SickGear Format
    @static_or_instance
    def sbfdate(self, dt=None, d_preset=None, setlocale=True):

        sbdatetime.setlocale(setlocale=setlocale)

        strd = ''
        try:
            obj = (dt, self)[self is not None]
            if None is not obj:
                strd = obj.strftime((sickbeard.DATE_PRESET, d_preset)[None is not d_preset])

        finally:
            sbdatetime.setlocale(setlocale=setlocale)
            return strd

    # display Datetime in SickGear Format
    @static_or_instance
    def sbfdatetime(self, dt=None, show_seconds=False, d_preset=None, t_preset=None, markup=False):

        sbdatetime.setlocale()

        strd = ''
        obj = (dt, self)[self is not None]
        try:
            if None is not obj:
                strd = u'%s, %s' % (obj.strftime((sickbeard.DATE_PRESET, d_preset)[None is not d_preset]),
                                    sbdatetime.sbftime(dt, show_seconds, t_preset, False, markup))

        finally:
            sbdatetime.setlocale(use_has_locale=sbdatetime.has_locale)
            return strd

    @static_or_instance
    def totimestamp(self, dt=None, default=None):
        obj = (dt, self)[self is not None]
        timestamp = default
        try:
            timestamp = time.mktime(obj.timetuple())
        finally:
            return (default, timestamp)[isinstance(timestamp, float)]