base/.vim/syntax/python/dateregex/dateregex/after.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3 # File:        after.py
   4 # Author:      Guilherme Victal <guilherme at victal.eti.br>
   5 # Description: Generates regexes after a certain date
   6 # License:     Vim license
   7 # Website:     http://github.com/freitass/todo.txt-vim
   8 # Version:     0.1
   9
  10 from datetime import date, timedelta, MAXYEAR
  11
  12
  13 def _year_regex_after(year):
  14     if int(year) > MAXYEAR:
  15         return None
  16
  17     year_regex = r'(\d+\d{%s}' % len(year)
  18     for idx, digit in enumerate(year):
  19         if digit != '9':
  20             regex = '|' + year[0:idx]
  21             regex += '9' if digit == '8' else '[%s-9]' % str(int(digit) + 1)
  22             if idx < len(year) - 1:
  23                 regex += '\d{%s}' % (len(year) - (idx + 1))
  24             year_regex += regex
  25
  26     year_regex += ')'
  27     return '-'.join((year_regex, r'\d{2}', r'\d{2}'))
  28
  29
  30 def _month_regex_after(year, month):
  31     if month == '12':
  32         return None
  33
  34     digit1, digit2 = month
  35     if digit1 == '1':
  36         month_regex = r'12' if month == '11' else r'1[12]'
  37     else:
  38         month_regex = r'1[0-2]'
  39         if digit2 != '9':
  40             if digit2 == '8':
  41                 month_regex = r'(' + month_regex + r'|09)'
  42             else:
  43                 month_regex = r'(' + month_regex + r'|0[%s-9])'
  44                 month_regex = month_regex % str(int(digit2) + 1)
  45     return '-'.join((year, month_regex, r'\d{2}'))
  46
  47 def _day_regex_after(year, month, day):
  48     last_month_day = str((date(int(year), (int(month) + 1) % 12, 1) + - date.resolution).day)
  49     if day == last_month_day:
  50         return None
  51     day_regex = r'('
  52     digit1, digit2 = day
  53     last_digit1, last_digit2 = last_month_day
  54     if digit1 == last_digit1:
  55         day_regex = last_month_day if int(digit2) == int(last_digit2) - 1 else last_digit1 + r'[%s-%s]' % (str(int(digit2) + 1), last_digit2)
  56     else:
  57         day_regex = r'('
  58         day_regex += last_digit1 if int(digit1) == int(last_digit1) - 1 else r'[%s-%s]' % (str(int(digit1) + 1), last_digit1)
  59         day_regex +=r'\d'
  60         if digit2 < '9':
  61             day_regex += '|' + digit1
  62             day_regex += '9' if digit2 == '8' else r'[%s-9]' % str(int(digit2) + 1)
  63
  64         day_regex += ')'
  65     return '-'.join((year, month, day_regex))
  66
  67
  68 def regex_date_after(given_date):
  69     year, month, day = given_date.isoformat().split('-')
  70
  71     year_regex = _year_regex_after(year)
  72     month_regex = _month_regex_after(year, month)
  73     day_regex = _day_regex_after(year, month, day)
  74
  75     date_regex = '(' + year_regex if year_regex else '('
  76     date_regex += ('|' + month_regex) if month_regex else ''
  77     date_regex += ('|' + day_regex) if day_regex else ''
  78     date_regex += ')'
  79     return date_regex
  80
  81
  82 def __main():
  83     import re
  84     date_regex = regex_date_after(date(1999,12,31))
  85     print(date_regex)
  86     pattern = re.compile(date_regex)
  87
  88
  89     d = date.today() + date.resolution
  90     assert pattern.match(date.strftime(d, '%Y-%m-%d')) is not None
  91     print(date.strftime(d, '%Y-%m-%d') + ' is okay')
  92     d += date.resolution
  93
  94 if __name__ == '__main__':
  95     __main()