diff --git a/cleanup_collection.py b/cleanup_collection.py index de4cad3..ceede06 100755 --- a/cleanup_collection.py +++ b/cleanup_collection.py @@ -33,9 +33,9 @@ for d in api.findDocuments({ if title != document['title']: edit['title'] = title - year = re.compile(' \d{4} ').findall(document['title']) - if year and year[0] in ('1', '2') and not document.get('date'): - document['date'] = edit['date'] = year[0] + year = re.compile(' (\d{4})( |$)').findall(document['title']) + if year and year[0][0][0] in ('1', '2') and not document.get('date'): + document['date'] = edit['date'] = year[0][0] if document.get('date') and document['date'] in title: title = title.replace(document['date'], '').strip()