欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python不规范的日期字符串处理类

程序员文章站 2022-06-07 08:35:22
我分析了形如19920203、199203、1992.02.03、1992.02、1992-02-03、1992-02、920203时间格式特征,列出了正则表达式如下:复制...

我分析了形如19920203、199203、1992.02.03、1992.02、1992-02-03、1992-02、920203时间格式特征,列出了正则表达式如下:

复制代码 代码如下:

^((?:19|20)?\d{2})[-.]?((?:[0-1]?|1)[0-9])[-.]?((?:[0-3]?|[1-3])[0-9])?$

当然这个表达式还不是很完善,只能做简单的切割,不能判断日期的合法性,关于日期是否合法,我还是交给python的时间功能来处理吧。

根据上面的正则表达式,我写的dateparser类如下:

复制代码 代码如下:

import re
import datetime

# ***************************************************
# *
# * description: 非标准的日期字符串处理
# * author: wangye  <pcn88 at hotmail dot com>
# *
# ***************************************************
class dateparser(object):

    def __init__(self):
        self.pattern = re.compile(
        r'^((?:19|20)?\d{2})[-.]?((?:[0-1]?|1)[0-9])[-.]?((?:[0-3]?|[1-3])[0-9])?$'
        )

    def __cutdate(self, date, flags):
        y = date.year
        m = date.month if flags[1] else 1
        d = date.day if flags[2] else 1
        return datetime.date(y, m, d)

    def __mergeflags(self, flags1, flags2):
        l = []
        length = min(len(flags1), len(flags2))
        for i in range(0, length):
            if flags1[i] and flags2[i]:
                l.append(true)
            else:
                l.append(false)
        return l

    def parse(self, strdate):
        """
        描述:时间解析方法。
        参数:strdate 要分析的时间字符串,比如目标时间类型datetime(1992, 2, 3)
              可以被解析的是下述字符串之一:
            19920203
            199203
            1992.02.03
            1992.02
            1992-02-03
            1992-02
            920203
        返回值:
            如果成功
            元组(datetime, flags),其中datetime表示转换完成的合法时间,
        flags是标志位,表示有效位数,比如199202实际转换了年和月,日没有,
        但是本函数将默认返回1日,但是flags将表示为(true, true, false),
        前面两个true分别表示年和月被转换,最后一个false表示日没有被转换。
            如果失败
            返回none。
        """
        m = self.pattern.match(strdate)
        flags = [false, false, false]
        if m:
            matches = list(m.groups())
            flags = list(map(lambda x:true if x!=none else false, matches))
            results = list(map(lambda x:int(x) if x!=none else 1, matches))
            # results = list(map(lambda x:1 if x==none else x, results))
            if results[0]<100:
                if results[0]>9:
                    results[0] += 1900
                else:
                    results[0] += 2000

            return (datetime.date(results[0], results[1], results[2]), flags)
        else:
            return none

    def convert(self, strdate, format):
        """
        描述:转换日期为指定格式。
        参数:strdate 同parse方法的strdate参数。
              format python时间格式标识,同datetime.date.strftime格式化标识。
        返回值:
            如果成功,返回指定format格式的时间字符串。
            如果失败,返回none。
        """
        date = self.parse(strdate)
        if date:
            date = date[0]
            return datetime.date.strftime(date, format)
        else:
            return none

    def compare(self, strdate1, strdate2):
        """
        描述:比较两个日期。
        参数:strdate1 和 strdate2 同parse方法的strdate参数
        返回值:
            可以是下列值之一
            -4  strdate1 无效,  strdate2 有效
            -3  strdate1 有效,  strdate2 无效
            -2  strdate1 和 strdate2 无效
            -1  strdate1 < strdate2
             0  strdate1 = strdate2
             1  strdate1 > strdate2
        """
        date1,flags1 = self.parse(strdate1)
        date2,flags2 = self.parse(strdate2)

        if date1 == none and date2 != none:
            return -4
        if date1 != none and date2 == none:
            return -3
        elif date1 == none and date2 == none:
            return -2

        flags = self.__mergeflags(flags1, flags2)
        date1 = self.__cutdate(date1, flags)
        date2 = self.__cutdate(date2, flags)

        if date1>date2:
            return 1
        elif date1<date2:
            return -1
        else:
            return 0

下面举几个例子供大家参考:

复制代码 代码如下:

>>> dateparser().parse("19860126")
(datetime.date(1986, 1, 26), [true, true, true])
>>> dateparser().parse("199111")
(datetime.date(1991, 11, 1), [true, true, false])
>>> dateparser().parse("1991")
(datetime.date(1919, 9, 1), [true, true, true])
>>> dateparser().parse("8511")
(datetime.date(1985, 11, 1), [true, true, false])
>>> dateparser().convert("19911101", "%y * %m * %d")
'1991 * 11 * 01'
>>> dateparser().convert("1990.1.01", "%y.%m.%d")
'1990.01.01'
>>> dateparser().compare("1992.2", "19922")
0
>>> dateparser().compare("1992.2", "1956.03.1")
1