pandora_leftovers/cleanup/add_volume.py

48 lines
1.3 KiB
Python
Executable File

#!/usr/bin/python3
import re
import sys
import ox
import ox.api
site = 'archive.leftove.rs'
api = ox.api.signin('https://%s/api/' % site)
if len(sys.argv) > 1:
collection = sys.argv[1]
conditions = [
{'key': 'collection', 'operator': '==', 'value': collection}
]
else:
conditions = []
for regex, fragments in (
['Vol. (\d+), No. (\d+)', ['Vol.', 'No.']],
['Vol (\d+) no (\d+)', [' Vol ', ' no ']],
['vol (\d+) no (\d+)', [' Vol ', ' no ']],
['Vol:(\d+) #(\d+)', ['Vol:', '#']],
):
query = {
'conditions': [
{'key': 'title', 'operator': '=', 'value': fragment}
for fragment in fragments
] + conditions,
'operator': '&'
}
for doc in api.findDocuments({
'query': query,
'keys': ['id', 'title', 'volume', 'issue'],
'range': [0, 10000]
})['data']['items']:
if not doc.get('volume', ''):
m = re.compile(regex).findall(doc['title'])
if m:
print(m[0], doc['id'], doc['title'])
volume = m[0][0]
issue = m[0][1]
api.editDocument({
'id': doc['id'],
'volume': volume,
'issue': issue,
})