[python] 最速日付解析ライブラリ ciso8601 のサンプル

ciso8601_example.py

"""
日付解析ライブラリ：ciso8601のサンプル
最速らしいので試してみた。
"""
import argparse
import time
from datetime import datetime, timedelta

import ciso8601


def main(args):
    epoch = datetime.utcfromtimestamp(0)
    datetimes = [
        (epoch + timedelta(seconds=i)).isoformat()
        for i in range(args.count)
    ]

    f = _call_ciso8601 if args.mode == 'ciso8601' else _call_datetime

    start = time.perf_counter()
    f(datetimes, args)
    elapsed = round(time.perf_counter() - start, 3)
    print(f'[{args.mode}]\tcount: {args.count}\telapsed: {elapsed}secs')  # noqa


def _call_ciso8601(datetimes, args):
    f = ciso8601.parse_datetime_unaware if args.call_unaware else ciso8601.parse_datetime  # noqa

    print(f'[method] {f.__name__}')
    for d in datetimes:
        f(d)


def _call_datetime(datetimes, args):
    for d in datetimes:
        datetime.strptime(d, '%Y-%m-%dT%H:%M:%S')


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('--count', type=int, default=100000)
    parser.add_argument('--mode', default='ciso8601', choices=['ciso8601', 'datetime'])  # noqa
    parser.add_argument('--call-unaware', action='store_true', default=False)
    args = parser.parse_args()
    main(args)

memo.md

参考情報

• ciso8601(github)
• Fastest Python datetime parser

以下のような結果となった。確かに速い。

通常の datetime.strptime の場合

$ python ./ciso8601_example.py --mode datetime
[datetime]      count: 100000   elapsed: 1.225secs

ciso8601

$ python ./ciso8601_example.py
[ciso8601]      count: 100000   elapsed: 0.022secs

naiveな日付の場合は、parse_datetime_unawareを使ったほうが速いとREADMEに記載があった。