87 lines
2.1 KiB
Python
Executable File
87 lines
2.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""Script for updating data. """
|
|
|
|
import argparse
|
|
import json
|
|
import os
|
|
import subprocess
|
|
from datetime import datetime, timedelta, tzinfo
|
|
|
|
from tqdm import tqdm
|
|
|
|
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: int) -> str:
|
|
"""Update and store data for a year.
|
|
|
|
Args:
|
|
year (int): Year
|
|
|
|
Returns:
|
|
str: Stored data path
|
|
"""
|
|
|
|
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():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--all', action='store_true')
|
|
args = parser.parse_args()
|
|
|
|
now = datetime.now(ChinaTimezone())
|
|
|
|
filenames = []
|
|
progress = tqdm(range(2014 if args.all else now.year, now.year + 2))
|
|
for i in progress:
|
|
progress.set_description(f'Updating {i} data')
|
|
filename = update_data(i)
|
|
filenames.append(filename)
|
|
|
|
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()
|