print 'ol.autologin() error; retrying'
conn = sqlite3.connect(db)
c = conn.cursor()
reader = csv.reader(open(csvfile), delimiter='\t', quotechar='|')
for row in reader:
ltid = row[0]
olid = row[1]
key = '/books' + olid[olid.rindex('/'):len(olid)]
c.execute('select * from ids where key = ?', (key, ))
x = c.fetchone()
if x != None:
continue
print 'Trying to get key: %r' % key
for attempt in range(5):
try:
data = ol.get(key)
break
except:
print 'ol.get() error; retrying'
if data.has_key('identifiers'):
if data['identifiers'].has_key('librarything'):
if data['identifiers']['librarything'].count(ltid) == 0:
data['identifiers']['librarything'].append(ltid)
else:
data['identifiers']['librarything'] = [ltid]
else:
data['identifiers'] = {'librarything': [ltid]}
for attempt in range(5):
try:
print ol.save(key, data, 'added LibraryThing ID')
break
'type': '/type/work',
'authors': None,
'title': None,
}
queue = []
for w in query_iter(work_q):
if not w.get('authors'):
print('no authors')
continue
if any(isinstance(a, dict) and 'author' in a for a in w['authors']):
continue
print(len(queue), w['key'],
w['title']) # , ol.get(w['authors'][0]['key'])['name']
full = ol.get(w['key'])
authors = full['authors']
assert all(isinstance(a, Reference) for a in authors)
full['authors'] = [{'author': a} for a in authors]
queue.append(full)
if len(queue) > 1000:
print('saving')
print(
ol.save_many(queue,
'update format of authors in works to provide roles'))
queue = []
print('two second pause')
sleep(2)
continue
work_e = {
'type': '/type/edition',
skip = 'elephantatwaldor00mira'
skip = 'sybasesqlserverp00paul'
skip = 'karmadunl00dunl'
skip = 'norbychronicles00asim'
skip = 'elizabethbarrett00fors'
skip = None
updates = []
cur.execute('select ia, editions, done, unmerge_count from merge')
for ia, ekeys, done, unmerge_count in cur.fetchall():
if skip:
if ia == skip:
skip = None
else:
continue
ekeys = ['/books/OL%dM' % x for x in sorted(int(re_edition_key.match(ekey).group(1)) for ekey in ekeys.split(' '))]
editions = [ol.get(ekey) for ekey in ekeys]
if any('authors' not in e or 'works' not in e for e in editions):
continue
author0 = editions[0]['authors'][0]
work0 = editions[0]['works'][0]
try:
if not all(author0 == e['authors'][0] for e in editions[1:]):
continue
except:
print('editions:', [e['key'] for e in editions])
raise
if all(work0 == e['works'][0] for e in editions[1:]):
continue
wkeys = []
for e in editions:
re_nonword = re.compile(r'\W', re.U)
conn = MySQLdb.connect(db='merge_editions')
cur = conn.cursor()
cur2 = conn.cursor()
ol = OpenLibrary('http://openlibrary.org/')
ol.login('EdwardBot', 'As1Wae9b')
cur.execute('select ia, editions, done from merge where done is null and unmerge_count=0')
for ia, ekeys, done in cur.fetchall():
updates = []
ekeys = ['/books/OL%dM' % x for x in sorted(int(re_edition_key.match(ekey).group(1)) for ekey in ekeys.split(' '))]
print((ia, ekeys))
min_ekey = ekeys[0]
editions = [ol.get(ekey) for ekey in ekeys]
master = editions[0]
for e in editions:
for k in 'classifications', 'identifiers', 'table_of_contents':
if k in e and not e[k]:
del e[k]
all_keys = set()
for e in editions:
all_keys.update(k for k, v in e.items() if v)
for k in 'latest_revision', 'revision', 'created', 'last_modified', 'key', 'type', 'genres':
if k in all_keys:
all_keys.remove(k)
for k in all_keys.copy():
saved = False
#for attempt in range(5):
try:
print 'Trying to save_many'
print ol.save_many(newlist, 'added LibraryThing ID')
saved = True
except KeyboardInterrupt:
sys.exit(0)
except:
badlist = []
for e in newlist:
for akey in e.get('authors', []):
got_author = False
for attempt in range(5):
try:
a = ol.get(akey['key'])
got_author = True
break
except:
print 'ol.get(author) error; retrying.'
if not got_author:
sys.exit('Failed to get author: %r' % akey['key'])
if a['type'] == '/type/author':
continue
if badlist.count(e) == 0:
badlist.append(e)
for badbook in badlist:
newlist.remove(badbook)
f.write(badbook['key'])
f.write('\n')
print ol.save_many(newlist, 'added LibraryThing ID')
from openlibrary.api import OpenLibrary
ol = OpenLibrary('http://openlibrary.org/')
data = eval(open('update').read())
done = []
for author in data:
print(author['key'], author['name'])
akey = author['key']
a = ol.get(akey)
if not a.get('bio') and author['bio']:
a['bio'] = author['bio']
ol.save(akey, a, 'Add author bio from Smashwords.')
for edition in author['editions']:
# wkey = ol.new({
# 'type': '/type/work',
# 'title': edition['title'],
# 'authors': [{'author': {'key': akey}}],
# 'description': edition['description'],
# 'subjects': ['Lending library'],
# })
#
wkey = edition['work']
w = ol.get(wkey)
assert edition['description']
if not w.get('description'):
w['description'] = edition['description']
if 'Lending library' not in w.get('subjects', []):
from openlibrary.catalog.utils.query import query, withKey
from openlibrary.api import OpenLibrary, unmarshal
from openlibrary.catalog.read_rc import read_rc
rc = read_rc()
ol = OpenLibrary("http://openlibrary.org")
ol.login('ImportBot', rc['ImportBot'])
to_fix = []
num = 0
for line in open('no_index'):
for e in query({'type': '/type/edition', 'title': None, 'ocaid': line[:-1]}):
num += 1
print num, e['key'], `e['title']`, line[:-1]
e2 = ol.get(e['key'])
del e2['ocaid']
to_fix.append(e2)
ol.save_many(to_fix, 'remove link')
rec_no = 0
chunk = 50
load_count = 0
archive_id = sys.argv[1]
def get_with_retry(key):
for i in range(3):
try:
return ol.get(key)
except urllib2.HTTPError, error:
if error.code != 500:
raise
print 'retry save'
sleep(10)
return ol.get(key)
def save_with_retry(key, data, comment):
for i in range(3):
try:
return ol.save(key, data, comment)
except urllib2.HTTPError, error:
if error.code != 500:
raise
print 'retry save'
sleep(10)
# urllib2.HTTPError: HTTP Error 500: Internal Server Error
def percent(a, b):
return float(a * 100.0) / b
work_q = {
'type': '/type/work',
'authors': None,
'title': None,
}
queue = []
for w in query_iter(work_q):
if not w.get('authors'):
print('no authors')
continue
if any(isinstance(a, dict) and 'author' in a for a in w['authors']):
continue
print(len(queue), w['key'], w['title']) # , ol.get(w['authors'][0]['key'])['name']
full = ol.get(w['key'])
authors = full['authors']
assert all(isinstance(a, Reference) for a in authors)
full['authors'] = [{'author':a} for a in authors]
queue.append(full)
if len(queue) > 1000:
print('saving')
print(ol.save_many(queue, 'update format of authors in works to provide roles'))
queue = []
print('two second pause')
sleep(2)
continue
work_e = {
'type': '/type/edition',
'works': w['key'],
'by_statement': None,
from openlibrary.api import OpenLibrary
ol = OpenLibrary('http://openlibrary.org/')
data = eval(open('update').read())
done = []
for author in data:
print (author['key'], author['name'])
akey = author['key']
a = ol.get(akey)
if not a.get('bio') and author['bio']:
a['bio'] = author['bio']
ol.save(akey, a, 'Add author bio from Smashwords.')
for edition in author['editions']:
# wkey = ol.new({
# 'type': '/type/work',
# 'title': edition['title'],
# 'authors': [{'author': {'key': akey}}],
# 'description': edition['description'],
# 'subjects': ['Lending library'],
# })
#
wkey = edition['work']
w = ol.get(wkey)
assert edition['description']
if not w.get('description'):
w['description'] = edition['description']
if 'Lending library' not in w.get('subjects', []):
print 'ol.autologin() error; retrying'
conn = sqlite3.connect(db)
c = conn.cursor()
reader = csv.reader(open(csvfile), delimiter='\t', quotechar='|')
for row in reader:
ltid = row[0]
olid = row[1]
key = '/books' + olid[olid.rindex('/'):len(olid)]
c.execute('select * from ids where key = ?', (key,))
x = c.fetchone()
if x != None:
continue
print 'Trying to get key: %r' % key
for attempt in range(5):
try:
data = ol.get(key)
break
except:
print 'ol.get() error; retrying'
if data.has_key('identifiers'):
if data['identifiers'].has_key('librarything'):
if data['identifiers']['librarything'].count(ltid) == 0:
data['identifiers']['librarything'].append(ltid)
else:
data['identifiers']['librarything'] = [ltid]
else:
data['identifiers'] = {'librarything': [ltid]}
for attempt in range(5):
try:
print ol.save(key, data, 'added LibraryThing ID')
break
ol.save_many(fix_redirects, "merge works")
all_existing = set()
work_keys = []
for w in works:
existing = set()
for e in w['editions']:
existing.update(edition_to_work[e])
if not existing: # editions have no existing work
if do_updates:
wkey = ol.new({'title': w['title'], 'type': '/type/work'})
print 'new work:', wkey, `w['title']`
#print 'new work:', `w['title']`
update = []
for ekey in w['editions']:
e = ol.get(ekey)
if do_updates:
e['works'] = [Reference(wkey)]
update.append(e)
if do_updates:
ol.save_many(update, "add editions to new work")
work_keys.append(wkey)
elif len(existing) == 1:
key = list(existing)[0]
work_keys.append(key)
if work_title[key] == w['title']:
print 'no change:', key, `w['title']`
else:
print 'new title:', key, `work_title[key]`, '->', `w['title']`
print sorted(work_to_edition[key])
print w['editions']
re_edition_key = re.compile('^/(?:books|b)/(OL\d+M)$')
archive_id = sys.argv[1]
def get_with_retry(key):
for i in range(3):
try:
return ol.get(key)
except urllib2.HTTPError, error:
if error.code != 500:
raise
print 'retry save'
sleep(10)
return ol.get(key)
def save_with_retry(key, data, comment):
for i in range(3):
try:
return ol.save(key, data, comment)
except urllib2.HTTPError, error:
if error.code != 500:
raise
print 'retry save'
sleep(10)
# urllib2.HTTPError: HTTP Error 500: Internal Server Error
class VacuumBot:
"""VacuumBot can help clean up Open Library, just tell it what to do!
The VacuumBot has (its) methods to do specific cleanup tasks.
It needs the credentials of a bot account on Open Library and some instructions.
Naturally, it shows no mercy.
"""
def __init__(self, username, password):
"""Takes a username and password of a bot account to establish a connection to OL.
"""
self.ol = OpenLibrary()
self.ol.login(username, password)
self.pagreg = re.compile(r"[^\s]\s+[:;]$")
self.emptypagreg = re.compile(r"[,.:;]+$")
self.formatdict = simplejson.load(codecs.open("formatdict.json", "rb", "utf-8"))
self.enc2 = codecs.getencoder("ascii")
self.savebuffer = {}
self.badrecords = []
self.aucache = {}
self.wocache = {}
#self.formatcache = NKCache("ol_books_formats", api_key = "cfdeaeda-4a22-4ae7-a2bf-1634da98fa1b")
self.logfile = codecs.EncodedFile(open("vacuumbot-log.tsv", "ab"), "unicode_internal", "utf-8", "replace")
def enc(self, str):
return self.enc2(str, "backslashreplace")[0]
def flog(self, key, operation, message):
"""Log to file 'vacuumbot-log.tsv'. Lines are time, key, operation and message, tab-separated.
"""
self.logfile.write(unicode(strftime("%Y-%m-%d_%H:%M:%S", localtime()) + "\t" + key + "\t" + operation + "\t" + message + "\n"))
def save_error(self, key, message):
errorfile = codecs.EncodedFile(open("vacuumbot-errors.txt", "ab"), "unicode_internal", "utf-8", "replace")
errorfile.write(unicode("[" + strftime("%Y-%m-%d_%H:%M:%S", localtime()) + "] Could not save record for: " + key + ", error was: " + message + "\n"))
errorfile.close()
def query(self, query):
return self.ol.query(query)
def ol_save(self, key, record, message):
try:
self.ol.save(key, record, self.enc(message))
self.flog(key, "direct save", message)
print_log("Saved "+key+": "+message)
except OLError as e:
self.save_error(key, str(e))
print_log("Save failed: "+str(e))
def ol_save2(self, key, record, message):
if message != None:
record = marshal(record)
if message in self.savebuffer.keys():
self.savebuffer[message][key] = record
if len(self.savebuffer[message]) >= 100:
self.flush(message)
else:
self.savebuffer[message] = {}
self.savebuffer[message][key] = record
self.flog(key, "buffer save", message)
else:
raise Exception("Message for saving is missing!")
def flush(self, buffer_name):
try:
if len(self.savebuffer[buffer_name]) > 0:
self.ol.save_many(self.savebuffer[buffer_name].values(), self.enc(buffer_name))
for key in self.savebuffer[buffer_name].keys():
self.flog(key, "buffer flush", buffer_name)
print_log("Flushed buffer ("+str(len(self.savebuffer[buffer_name]))+" records): "+buffer_name)
self.savebuffer[buffer_name] = {}
sleep(1)
except OLError as e:
# Try to remove rejected record from buffer
err_mess = simplejson.loads(re.sub(r'^[^{]*', "", str(e)))
if err_mess["error"] == "bad_data":
k = err_mess["at"]["key"]
del self.savebuffer[buffer_name][k]
self.save_error(k, "Multisave failed: "+str(e)+"; removed record from buffer")
else:
k = self.savebuffer[buffer_name].keys()[0]
self.save_error(k, "Multisave failed: "+str(e))
def flush_all(self):
for m in self.savebuffer.keys():
self.flush(m)
def ol_get(self, key, v=None):
"""Gets a record from OL and catches OLErrors.
Make sure you check for None when you process this function's result.
"""
try:
return self.ol.get(key, v)
except OLError as e:
self.save_error(key, str(e))
print_log("Get failed: "+str(e))
def clean_author_dates(self):
for year in range(1900, 2013):
authors = self.query({"type": "/type/author", "death_date": str(year)+".", "limit": False})
print_log("Getting authors with death date '" + str(year) + "'...")
for author in authors:
obj = self.ol_get(author)
self.clean_author(obj)
sleep(2)
done = codecs.EncodedFile(open("cleanauthors-done.txt", "ab"), "unicode_internal", "utf-8", "replace")
done.write("Death date '" + str(year) + ".' updated to '" + str(year) + "'\n")
done.close()
def clean_author_dates2(self):
for year in range(0,1000):
# Get keys of all authors with death date <x>
authors = self.query({"type": "/type/author", "death_date": str(year)+".", "limit": False})
print_log("Getting authors with death date '" + str(year) + "'...")
list = []
for author in authors:
# Feed authors to buffer list
list.append(author)
if len(list) > 99:
# Process these few authors before continuing feeding
# Get records
print_log("Getting full records")
records = self.ol.get_many(list)
for obj in records.itervalues():
self.clean_author2(obj)
list = []
if len(list) > 0:
records = self.ol.get_many(list)
for obj in records.itervalues():
self.clean_author2(obj)
self.flush_all()
done = codecs.EncodedFile(open("cleanauthors-done.txt", "ab"), "unicode_internal", "utf-8", "replace")
done.write(unicode("Death date '" + str(year) + ".' updated to '" + str(year) + "'\n"))
done.close()
sleep(0.5)
def clean_author(self, obj):
"""Clean author records. For example removes the period after the death date.
Saves the updated record directly, instead of putting it in the save buffer.
"""
# Remove period from death date
comment = []
result1 = self.clean_death_date(obj)
if result1 != None:
comment.append("Removed period from death date")
if len(comment) > 0:
self.ol_save(obj["key"], result1, "; ".join(comment))
def clean_author2(self, obj):
"""Clean author records. For example removes the period after the death date.
Saves the updated record in the save buffer.
"""
# Remove period from death date
comment = []
result1 = self.clean_death_date(obj)
if result1 != None:
comment.append("Removed period from death date")
if len(comment) > 0:
self.ol_save2(obj["key"], result1, "; ".join(comment))
def clean_death_date(self, obj):
changed = False
if "death_date" in obj.keys():
if re.match(r"^\d{1,4}\.", obj["death_date"]):
obj["death_date"] = obj["death_date"].rstrip(" .")
changed = True
elif obj["death_date"] == "":
del obj["death_date"]
changed = True
if changed:
return obj
else:
return None
def update_author_in_edition(self):
aucache = {}
for line in open("errors3.txt", "rb"):
error = simplejson.loads(line)
if error["error"] == "bad_data":
olid = error["at"]["key"]
try:
obj = self.ol_get(olid)
print "Edition found and downloaded"
if "works" in obj.keys() and len(obj["works"]) > 0:
# Get work, see if it has at least one author
work = self.ol_get(obj["works"][0])
print "Work found and downloaded"
if "authors" in work.keys() and len(work["authors"]) > 0:
del obj["authors"]
self.ol_save(olid, obj, "Removed author from Edition (author found in Work)")
else:
# Get author and find new author key
if "authors" in obj.keys() and len(obj["authors"]) > 0:
newau = []
for auID in obj["authors"]:
if auID in aucache.keys():
# already looked up and stored in cache
newau.append(aucache[auID])
print "new ID found in cache"
else:
# lookup author's new ID
newID = self.find_new_author(auID)
aucache[auID] = newID
newau.append(newID)
obj["authors"] = newau
self.ol_save(olid, obj, "Updated author in Edition (author was merged but not updated here)")
except:
print "Error trying to fix", olid
def find_new_author(self, olid):
obj = self.ol_get(olid)
print "downloaded", olid
if obj["type"] == "/type/author":
return obj["key"]
elif obj["type"] == "/type/redirect":
return self.find_new_author(obj["location"])
elif obj["type"] == "/type/delete":
# author was deleted, undelete
obj["type"] = "/type/author"
self.ol_save(obj["key"], obj, "Undeleted record, because other records still referred to it")
return obj["key"]
def clean_physical_object(self, obj):
"""Cleans up physical aspects of the given Edition object, such as format and pagination. Returns the cleaned obj.
Physical format: calls self.clean_format(obj).
Pagination: calls self.clean_pagination(obj).
"""
obj1 = self.clean_format(obj)
if obj1 != None:
obj2 = self.clean_pagination(obj1)
else:
obj2 = self.clean_pagination(obj)
if obj2 != None:
return obj2
else:
return obj
def clean_format(self, obj):
"""Cleans up the obj's physical_format field, removing it from obj if necessary.
Checks the current value of physical_format, looks up the replacement value and updates if new value is different from current value. If the format field is empty after this, it is removed.
If nothing changed or there is no physical_format field to begin with, None is returned.
"""
if "physical_format" in obj.keys():
if obj["physical_format"] == "":
# Remove empty field and return
del obj["physical_format"]
return obj
else:
# Check if there is a better format
v = self._check_format(obj["physical_format"])
if v != "":
# Use new value
obj["physical_format"] = v
return obj
elif v == "":
# New value would leave empty field -> remove field
del obj["physical_format"]
return obj
else:
return None
else:
return None
def _check_format(self, format):
# if format matches punctuation regular expression, create normalized format string before lookup
checkstr = re.sub(r"", "", format)
# and look up replacement format in dictionary
if checkstr in self.formatdict.keys() and self.formatdict[checkstr] != format:
return self.formatdict[checkstr]
else:
# if there is no new value or new is same as original, don't update.
return False
def replace_formats(self, old, new):
"""Replaces the old value in physical format fields by the new value.
This method tries to process all records with old as format value, which are potentially millions of records.
"""
print_log("Getting records with format '"+old+"'...")
olids = self.ol.query({"type":"/type/edition", "physical_format": old, "limit": False})
for r in olids:
print "Improving", r
self.replace_format(r, old, new)
sleep(4)
def replace_formats_clean_pagination(self, old, new):
"""Replaces the old value in physical format fields by the new value.
This method tries to process all records with old as format value, which are potentially millions of records.
"""
print_log("Getting records with format '"+old+"'...")
olids = self.ol.query({"type":"/type/edition", "physical_format": old, "limit": False})
for olid in olids:
gotit = False
tries = 0
while (not gotit and tries <= 5):
obj = self.ol_get(olid)
if obj:
gotit = True
else:
sleep(10)
tries = tries + 1
if not gotit:
raise Exception("timeout")
comment = []
print "Improving", olid
# Step 1: replace format
result1 = self.replace_format2(obj, old, new)
if result1:
comment.append("Updated format '"+old+"' to '"+new+"'")
else:
result1 = obj
# Step 2: Use Step 1 output
result2 = self.clean_pagination(result1)
if result2:
comment.append("cleaned up pagination")
else:
result2 = result1
# Something changed if comment is not empty
if len(comment) != 0:
self.ol_save(obj["key"], result2, "; ".join(comment))
print_log("; ".join(comment))
else:
print_log("Did nothing, really.")
sleep(3)
def replace_split_formats_clean_pagination(self, old, new, by, sub, ot):
"""Replaces the old value in physical format fields by the new value and puts the by statement in the correct place.
This method tries to process 1000 records with old as format value, which are potentially millions of records.
"""
print_log("Getting records with format '"+old+"'...")
olids = self.ol.query({"type":"/type/edition", "physical_format": old, "limit": 1000})
list = []
for olid in olids:
# Feed authors to buffer list
list.append(olid)
if len(list) > 99:
# Process these few authors before continuing feeding
# Get records
print_log("Getting full records")
records = self.ol.get_many(list)
for obj in records.itervalues():
obj = unmarshal(obj)
self._replace_split_formats_clean_pagination(obj, old, new, by, sub, ot)
list = []
if len(list) > 0:
records = self.ol.get_many(list)
for obj in records.itervalues():
obj = unmarshal(obj)
self._replace_split_formats_clean_pagination(obj, old, new, by, sub, ot)
self.flush_all()
def replace_formats_clean_pagination2(self, old, new):
"""Replaces the old value in physical format fields by the new value.
This method tries to process all records with old as format value, which are potentially millions of records.
"""
print_log("Getting records with format '"+old+"'...")
olids = self.ol.query({"type":"/type/edition", "physical_format": old, "limit": 1000})
list = []
for olid in olids:
# Feed authors to buffer list
list.append(olid)
if len(list) > 99:
# Process these few authors before continuing feeding
# Get records
print_log("Getting full records")
records = self.ol.get_many(list)
for obj in records.itervalues():
obj = unmarshal(obj)
self._replace_formats_clean_pagination(obj, old, new)
list = []
if len(list) > 0:
records = self.ol.get_many(list)
for obj in records.itervalues():
obj = unmarshal(obj)
self._replace_formats_clean_pagination(obj, old, new)
self.flush_all()
def _replace_formats_clean_pagination(self, obj, old, new):
comment = []
# Step 1: replace format
result1 = self.replace_format2(obj, old, new)
if result1[1]:
comment.append(result1[1])
# Step 2: Use Step 1 output
result2 = self.clean_pagination(result1[0])
if result2[1]:
comment.append(result2[1])
# Step 3: use Step 2 output
result3 = self._update_author_in_edition(result2[0])
if result3[1]:
comment.append(result3[1])
# Something changed if comment is not empty
if len(comment) != 0:
self.ol_save2(obj["key"], result3[0], "; ".join(comment))
def _replace_split_formats_clean_pagination(self, obj, old, new, by):
comment = []
# Step 1: replace format
result1 = self.replace_format2(obj, old, new)
if result1[1]:
comment.append(result1[1])
# Step 1a: add the By statement
result1a = self.add_by(result1[0], by)
if result1a[1]:
comment.append(result1a[1])
# Step 2: Use Step 1a output
result2 = self.clean_pagination(result1a[0])
if result2[1]:
comment.append(result2[1])
# Step 3: use Step 2 output
result3 = self._update_author_in_edition(result2[0])
if result3[1]:
comment.append(result3[1])
# Something changed if comment is not empty
if len(comment) != 0:
self.ol_save2(obj["key"], result3[0], "; ".join(comment))
def _replace_split_formats_clean_pagination(self, obj, old, new, by, sub, ot):
comment = []
# Step 1: replace format
result1 = self.replace_format2(obj, old, new)
if result1[1]:
comment.append(result1[1])
# Step 1a: add the By statement
result1a = self.add_by(result1[0], by)
if result1a[1]:
comment.append(result1a[1])
# Step 1b: add the subtitle
result1b = self.add_subtitle(result1a[0], sub)
if result1b[1]:
comment.append(result1b[1])
# Step 1c: add the other_titles
result1c = self.add_subtitle(result1b[0], ot)
if result1c[1]:
comment.append(result1c[1])
# Step 2: Use Step 1b output
result2 = self.clean_pagination(result1b[0])
if result2[1]:
comment.append(result2[1])
# Step 3: use Step 2 output
result3 = self._update_author_in_edition(result2[0])
if result3[1]:
comment.append(result3[1])
# Something changed if comment is not empty
if len(comment) != 0:
self.ol_save2(obj["key"], result3[0], "; ".join(comment))
def add_by(self, obj, by):
if by != "":
if "by_statement" in obj.keys():
if len(obj["by_statement"]) == 0:
obj["by_statement"] = by
return (obj, "updated By statement")
elif "notes" in obj.keys(): # and "value" in obj["notes"].keys()
print self.enc(obj["notes"])
if isinstance(obj["notes"], Text):
obj["notes"] = obj["notes"] + "\n\nBy statement found in format: " + by + "\n"
return (obj, "added By statement to notes, because By statement field was not empty")
elif isinstance(obj["notes"], dict):
d = unmarshal(obj["notes"])
obj["notes"] = d + "\n\nBy statement found in format: " + by + "\n"
return (obj, "added By statement to notes, because By statement field was not empty")
else:
obj["notes"] = Text("By statement found in format: " + by + "\n")
return (obj, "put By statement in notes")
else:
obj["notes"] = Text("By statement found in format: " + by + "\n")
return (obj, "put By statement in notes")
else:
obj["by_statement"] = by
return (obj, "added By statement")
else:
return (obj, None)
def add_subtitle(self, obj, sub):
if sub != "":
if "subtitle" in obj.keys():
if len(obj["subtitle"]) == 0:
obj["subtitle"] = sub
return (obj, "updated subtitle")
elif "notes" in obj.keys():
print self.enc(obj["notes"])
obj["notes"] = obj["notes"] + "\n\nSubtitle found in format: " + sub + "\n"
return (obj, "added subtitle to notes, because subtitle field was not empty")
else:
obj["notes"] = sub
return (obj, "put subtitle in notes")
else:
obj["subtitle"] = sub
return (obj, "added subtitle")
else:
return (obj, None)
def add_other_title(self, obj, ot):
if ot != "":
if "other_titles" in obj.keys():
if isinstance(obj["other_titles"], list):
obj["other_titles"].append(ot)
return (obj, "added title to other_titles")
elif isinstance(obj["other_titles"], string) or isinstance(obj["other_titles"], unicode):
obj["other_titles"] = [obj["other_titles"], ot]
return (obj, "changed other_titles from string to list of strings")
else:
raise Exception("other_titles is not a list or string")
else:
obj["other_titles"] = [ot]
return (obj, "created other_titles")
else:
return (obj, None)
def _update_author_in_edition(self, obj):
#try:
workhasauthors = False
hasauthors = "authors" in obj.keys() and len(obj["authors"]) > 0
if not hasauthors:
return (obj, None)
elif "works" in obj.keys() and len(obj["works"]) > 0:
#print obj["works"][0]
wID = obj["works"][0]
if wID in self.wocache.keys():
print "Work ID found in cache", wID
workhasauthors = self.wocache[wID]
else:
# Get work, see if it has at least one author
work = self.ol_get(wID)
print "Work found and downloaded", wID
self.wocache[wID] = "authors" in work.keys() and len(work["authors"]) > 0
workhasauthors = self.wocache[wID]
if workhasauthors and "authors" in obj.keys():
del obj["authors"]
return (obj, "Removed author from Edition (author found in Work)")
else:
# Edition has authors, Work has no authors. Work should be updated and then authors removed from edition.
# for now, update authors in edition (that causes error). Leave rest to WorkBot.
return self._replace_authors(obj)
else:
return self._replace_authors(obj)
#except Exception as e:
# print "Error trying to fix", obj["key"]
#print sys.exc_info()
def _replace_authors(self, obj):
newau = []
for auID in obj["authors"]:
if auID in self.aucache.keys():
# already looked up and stored in cache
newau.append(Reference(self.aucache[auID]))
print "new ID found in cache"
else:
# lookup author's new ID
newID = self.find_new_author(auID)
self.aucache[auID] = newID
newau.append(Reference(newID))
print obj["authors"]
print newau
if obj["authors"] != newau:
obj["authors"] = newau
comment = "replaced author(s) in Edition (reference was outdated)"
return (obj, comment)
else:
return (obj, None)
def replace_format(self, olid, old, new):
"""Replaces a value from the physical format field.
"""
try:
obj = self.ol.get(olid)
except OLError as e:
self.save_error(olid, str(e))
return
if "physical_format" in obj.keys() and obj["physical_format"] == old:
obj["physical_format"] = new
print_log("updating format for "+olid)
self.ol_save(obj["key"], obj, "Updated format '"+old+"' to '"+new+"'")
def replace_format2(self, obj, old, new):
"""Replaces a value from the physical format field.
Returns a tuple (obj, comment). comment is None if nothing changed. obj is updated or unchanged input obj.
"""
if "physical_format" in obj.keys() and obj["physical_format"] == old:
obj["physical_format"] = new
print_log("updating format for "+obj["key"])
return (obj, "Updated format '"+old+"' to '"+new+"'")
elif "physical_format" in obj.keys() and obj["physical_format"] == "":
del obj["physical_format"]
return (obj, "Deleted empty physical format field")
else:
return (obj, None)
def clean_pagination(self, obj):
"""Removes spaces, semicolons and colons from the end of the pagination field of obj.
If the pagination field is empty or only contains a combination of ',', '.', ';' and ':', the field is removed.
Returns a tuple of the possibly updated obj and a comment (which is None if nothing changed).
"""
if "pagination" in obj.keys():
# Strip commas, semicolons, colons, forward slashes and spaces from the right.
new = obj["pagination"].rstrip(" ,;:/")
if obj["pagination"] == new:
# Pagination did not change.
return (obj, None)
elif obj["pagination"] == "" or new == "" or self.emptypagreg.match(obj["pagination"]):
# field is empty, or only has ignorable characters, remove the field.
del obj["pagination"]
return (obj, "deleted empty pagination field")
else:
obj["pagination"] = new
return (obj, "cleaned up pagination")
else:
# There is no pagination field; return None.
return (obj, None)
def remove_classification_value(self, obj, type, value):
"""Removes a value from the list of <type> classifications.
For example, can be used to remove the "B" value from
Dewey Decimal classifications.
If the classifications list is empty afterwards, it is removed.
If the classifications object in the record is empty (because
removing the deleted list was the only one in it), it is removed
as well.
"""
special = ["lc_classifications", "dewey_decimal_class"]
if type in special and type in obj.keys():
while value in obj[type]:
obj[type].remove(value)
if len(obj[type]) == 0:
del obj[type]
elif "classifications" in obj.keys() and type in obj["classifications"].keys():
while value in obj["classifications"][type]:
obj["classifications"][type].remove(value)
if len(obj["classifications"][type]) == 0:
del obj["classifications"][type]
if len(obj["classifications"]) == 0:
del obj["classifications"]
def deduplicate_list(self, li):
"""Sorts a list and removes duplicate values in place."""
a = len(li)
c = 0
li.sort()
while c < a-1:
if li[c] == li[c+1]:
li.pop(c+1)
a = a-1
else:
c = c+1
def dedup(self, obj):
"""Removes duplicate values from an object.
Calls deduplicate_list for lists.
Calls itself on compound objects.
Does nothing with strings or other types.
"""
if isinstance(obj, str):
return
elif isinstance(obj, dict):
for k in obj:
dedup(obj[k])
elif isinstance(obj, list):
deduplicate_list(obj)
else:
return
def remove_key(self, olid, key):
"""Removes a key from a record
Use with caution :)
"""
object = ol.get(olid)
if key in object:
del object[key]
self.ol_save(object['key'], object, "Sucked up \"" + key + "\".")
def deduplicate_values(self, olid, key):
"""Removes duplicate values
Reads the values of a key and removes duplicate values,
leaving 1.
"""
object = ol.get(olid)
if key in object:
dedup(object[key])
def remove_classification(self, obj, classification):
if "classifications" in obj:
if classification in obj["classifications"]:
del obj["classifications"][classification]
def clean_lccn_permalink(self, olid):
"""Removes lccn_permalink from classifications
Removes permalink from classifications and adds the LCCN to
the identifiers, if is isn't there already.
"""
object = ol.get(olid)
if "classifications" in object.keys():
if "lccn_permalink" in object["classifications"].keys():
lccnst = []
for a in object["classifications"]["lccn_permalink"]:
b = a.rsplit("/",1)
lccnst.append(b[len(b)])
if "lccn" not in object.keys():
object["lccn"] = []
for l in lccnst:
if l not in object["lccn"]:
object["lccn"].append(l)
remove_classification(object, "lccn_permalink")
def vacuum(self, filename):
"""Main execution
Vacuums the Open Library based on commands found in the file.
Commands understood by VacuumBot are:
* remove_key
Command files are structured as follows: [todo]
"""
n = 0
for line in open(filename):
olid, isbn, goodreads_id = line.strip().split()
n = n+1
if (n % 100000) == 0:
print_log("(just read line " + str(n) + " from the command file)")
is_good = False
while (not is_good):
try:
map_id(olid, isbn, goodreads_id)
is_good = True
except:
print_log("Exception for Goodreads ID \"" + goodreads_id + "\", message: \"" + str(sys.exc_info()[1]) + "\"")
sleep(10)
skip = 'elizabethbarrett00fors'
skip = None
updates = []
cur.execute('select ia, editions, done, unmerge_count from merge')
for ia, ekeys, done, unmerge_count in cur.fetchall():
if skip:
if ia == skip:
skip = None
else:
continue
ekeys = [
'/books/OL%dM' % x for x in sorted(
int(re_edition_key.match(ekey).group(1))
for ekey in ekeys.split(' '))
]
editions = [ol.get(ekey) for ekey in ekeys]
if any('authors' not in e or 'works' not in e for e in editions):
continue
author0 = editions[0]['authors'][0]
work0 = editions[0]['works'][0]
try:
if not all(author0 == e['authors'][0] for e in editions[1:]):
continue
except:
print 'editions:', [e['key'] for e in editions]
raise
if all(work0 == e['works'][0] for e in editions[1:]):
continue
wkeys = []
for e in editions:
from openlibrary.catalog.utils.query import query, withKey
from openlibrary.api import OpenLibrary, unmarshal
from openlibrary.catalog.read_rc import read_rc
rc = read_rc()
ol = OpenLibrary("http://openlibrary.org")
ol.login('ImportBot', rc['ImportBot'])
to_fix = []
num = 0
for line in open('no_index'):
for e in query({
'type': '/type/edition',
'title': None,
'ocaid': line[:-1]
}):
num += 1
print(num, e['key'], repr(e['title']), line[:-1])
e2 = ol.get(e['key'])
del e2['ocaid']
to_fix.append(e2)
ol.save_many(to_fix, 'remove link')
ol = OpenLibrary('http://openlibrary.org/')
ol.login('EdwardBot', 'As1Wae9b')
cur.execute(
'select ia, editions, done from merge where done is null and unmerge_count=0'
)
for ia, ekeys, done in cur.fetchall():
updates = []
ekeys = [
'/books/OL%dM' % x for x in sorted(
int(re_edition_key.match(ekey).group(1))
for ekey in ekeys.split(' '))
]
print((ia, ekeys))
min_ekey = ekeys[0]
editions = [ol.get(ekey) for ekey in ekeys]
master = editions[0]
for e in editions:
for k in 'classifications', 'identifiers', 'table_of_contents':
if k in e and not e[k]:
del e[k]
all_keys = set()
for e in editions:
all_keys.update(k for k, v in e.items() if v)
for k in 'latest_revision', 'revision', 'created', 'last_modified', 'key', 'type', 'genres':
if k in all_keys:
all_keys.remove(k)
for k in all_keys.copy():
saved = False
#for attempt in range(5):
try:
print 'Trying to save_many'
print ol.save_many(newlist, 'added LibraryThing ID')
saved = True
except KeyboardInterrupt:
sys.exit(0)
except:
badlist = []
for e in newlist:
for akey in e.get('authors', []):
got_author = False
for attempt in range(5):
try:
a = ol.get(akey['key'])
got_author = True
break
except:
print 'ol.get(author) error; retrying.'
if not got_author:
sys.exit('Failed to get author: %r' % akey['key'])
if a['type'] == '/type/author':
continue
if badlist.count(e) == 0:
badlist.append(e)
for badbook in badlist:
newlist.remove(badbook)
f.write(badbook['key'])
f.write('\n')
print ol.save_many(newlist, 'added LibraryThing ID')