masterscraper/masterscraper/core/meta_year.py

32 lines
991 B
Python

#!/usr/bin/env python3
from datetime import date
import re
#--------[ Get Variable Year ]--------#
def meta_year(self):
key_year = []
for key in self.data[0]:
if re.match('\d\d\d\d', key):
key_year.append( key )
elif 'year' in self.info['keys']:
y1 = self.data[1][self.info['keys'].index('year')]
y2 = self.data[-1][self.info['keys'].index('year')]
if y1 <= y2: key_year.append( '{0}-{1}'.format(y1,y2) )
if y1 > y2: key_year.append( '{0}-{1}'.format(y2,y1) )
elif 'date' in self.info['keys']:
y1 = self.data[1][self.info['keys'].index('date')].split('-')[0]
y2 = self.data[-1][self.info['keys'].index('date')].split('-')[0]
if y1 <= y2: key_year.append( '{0}-{1}'.format(y1,y2) )
if y1 > y2: key_year.append( '{0}-{1}'.format(y2,y1) )
else:
key_year.append( date.today().strftime('%Y') )
self.info['year'] = key_year