66 lines
1.6 KiB
Python
Executable File
66 lines
1.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import json
|
|
import os
|
|
import subprocess
|
|
from datetime import datetime, timedelta, tzinfo
|
|
|
|
from fetch_holidays import CustomJSONEncoder, fetch_holiday
|
|
|
|
|
|
class ChinaTimezone(tzinfo):
|
|
"""Timezone of china. """
|
|
|
|
def tzname(self, dt):
|
|
return 'UTC+8'
|
|
|
|
def utcoffset(self, dt):
|
|
return timedelta(hours=8)
|
|
|
|
def dst(self, dt):
|
|
return timedelta()
|
|
|
|
|
|
__dirname__ = os.path.abspath(os.path.dirname(__file__))
|
|
|
|
|
|
def _file_path(*other):
|
|
|
|
return os.path.join(__dirname__, *other)
|
|
|
|
|
|
def update_data(year):
|
|
filename = _file_path(f'{year}.json')
|
|
with open(filename, 'w', encoding='utf-8', newline='\n') as f:
|
|
json.dump(fetch_holiday(year), f,
|
|
indent=4,
|
|
ensure_ascii=False,
|
|
cls=CustomJSONEncoder)
|
|
return filename
|
|
|
|
|
|
def main():
|
|
now = datetime.now(ChinaTimezone())
|
|
|
|
filenames = []
|
|
filenames.append(update_data(now.year))
|
|
filenames.append(update_data(now.year + 1))
|
|
|
|
subprocess.run(['git', 'add', *filenames], check=True)
|
|
diff = subprocess.run(['git', 'diff', '--stat', '--cached'],
|
|
check=True,
|
|
stdout=subprocess.PIPE,
|
|
encoding='utf-8').stdout
|
|
if not diff:
|
|
print('Already up to date.')
|
|
return
|
|
|
|
subprocess.run(
|
|
['git', 'commit', '-m', 'Update data [skip ci]\n\n' + diff], check=True)
|
|
subprocess.run(['git', 'tag', now.strftime('%Y.%m.%d')], check=True)
|
|
subprocess.run(['git', 'push'], check=True)
|
|
subprocess.run(['git', 'push', '--tags'], check=True)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|