def main(): options, args = parse_args() if options.src.startswith("http://"): src = OpenLibrary(options.src) else: src = Disk(options.src) if options.dest.startswith("http://"): dest = OpenLibrary(options.dest) section = "[%s]" % web.lstrips(options.dest, "http://").strip("/") if section in read_lines(os.path.expanduser("~/.olrc")): dest.autologin() else: dest.login("admin", "admin123") else: dest = Disk(options.dest) for list_key in options.lists: copy_list(src, dest, list_key, comment=options.comment) keys = args keys = list(expand(src, keys)) copy(src, dest, keys, comment=options.comment, recursive=options.recursive)
def main(): options, args = parse_args() if options.src.startswith("http://"): src = OpenLibrary(options.src) else: src = Disk(options.src) if options.dest.startswith("http://"): dest = OpenLibrary(options.dest) section = "[%s]" % web.lstrips(options.dest, "http://").strip("/") if section in read_lines(os.path.expanduser("~/.olrc")): dest.autologin() else: dest.login("admin", "admin123") else: dest = Disk(options.dest) for list_key in options.lists: copy_list(src, dest, list_key, comment=options.comment) keys = args keys = list(expand(src, keys)) copy(src, dest, keys, comment=options.comment, recursive=options.recursive)
def add_cover_to_work(w): if 'cover_edition' in w: return q = {'type':'/type/edition', 'works':w['key'], 'publish_date': None, 'languages': '/l/eng'} cover_edition = pick_cover(query_iter(q)) if not cover_edition: q = {'type':'/type/edition', 'works':w['key'], 'publish_date': None} cover_edition = pick_cover(query_iter(q)) if not cover_edition: return w['cover_edition'] = Reference(cover_edition) if ol is None: rc = read_rc() ol = OpenLibrary("http://openlibrary.org") ol.login('WorkBot', rc['WorkBot']) print ol.save(w['key'], w, 'added cover to work')
parser = argparse.ArgumentParser(description='scribe loader') parser.add_argument('--skip_hide_books', action='store_true') parser.add_argument('--item_id') parser.add_argument('--config', default='openlibrary.yml') args = parser.parse_args() config_file = args.config config.load(config_file) import_bot_password = config.runtime_config['load_scribe'][ 'import_bot_password'] # '/1/var/log/openlibrary/load_scribe' load_scribe_log = config.runtime_config['load_scribe']['log'] ol = OpenLibrary("http://openlibrary.org") ol.login('ImportBot', import_bot_password) password = Popen(["/opt/.petabox/dbserver"], stdout=PIPE).communicate()[0] db = web.database(dbn='mysql', host='dbmeta.us.archive.org', user='******', \ passwd=password, db='archive') db.printing = False re_census = re.compile('^\d+(st|nd|rd|th)census') re_edition_key = re.compile('^/(?:books|b)/(OL\d+M)$') def read_short_title(title): return str(fast_parse.normalize_str(title)[:25])
help="The openlibrary API host") op.add_option("-k", "--nyt-api-key", dest="nyt_api_key", help="API key for use with the nyt bestsellers api") op.add_option("-u", "--bot-username", dest="username", default="nyt_bestsellers_bot", help="The bot username for accessing the Open Library API") op.add_option("-p", "--bot-password", dest="password", help="The bot password for accessing the Open Library API") options, _ = op.parse_args() global NYT_API_KEY NYT_API_KEY = options.nyt_api_key global OL OL = OpenLibrary("http://%s" % options.openlibrary_host) OL.login(options.username, options.password) results = collections.defaultdict(list) for ln in get_nyt_bestseller_list_names(): LOG("INFO", "processing list %s" % ln) for i, book in enumerate(load_nyt_bestseller_list(ln)): ol_keys = reconcile_book(book) if not ol_keys: LOG("WARN", "unable to reconcile '%s' by %s - no OL book found" % ( book['book_details'][0]['title'], book['book_details'][0]['author'] )) if not (key for key in ol_keys if key.startswith("/works/")): LOG("WARN", "only editions for '%s' by %s: %s" % ( book['book_details'][0]['title'], book['book_details'][0]['author'], ol_keys )) results[ln].append({ "nyt": book,
from openlibrary.catalog.read_rc import read_rc import openlibrary.catalog.merge.amazon as amazon from openlibrary.catalog.get_ia import * from openlibrary.catalog.importer.db_read import withKey, get_mc from openlibrary.api import OpenLibrary, Reference import openlibrary.catalog.marc.fast_parse as fast_parse import xml.parsers.expat import web, sys from time import sleep import six rc = read_rc() ol = OpenLibrary("http://openlibrary.org") ol.login('ImportBot', rc['ImportBot']) ia_db = web.database(dbn='mysql', db='archive', user=rc['ia_db_user'], pw=rc['ia_db_pass'], host=rc['ia_db_host']) ia_db.printing = False re_meta_marc = re.compile('([^/]+)_(meta|marc)\.(mrc|xml)') threshold = 875 amazon.set_isbn_match(225) def try_amazon(thing):
from __future__ import print_function import MySQLdb, datetime, re, sys sys.path.append('/1/src/openlibrary') from openlibrary.api import OpenLibrary, Reference from pprint import pprint import six conn = MySQLdb.connect(db='merge_editions') cur = conn.cursor() re_edition_key = re.compile('^/books/OL(\d+)M$') re_work_key = re.compile('^/works/OL(\d+)W$') ol = OpenLibrary('http://openlibrary.org/') ol.login('EdwardBot', 'As1Wae9b') re_iso_date = re.compile('^(\d{4})-\d\d-\d\d$') re_end_year = re.compile('(\d{4})$') def get_publish_year(d): if not d: return m = re_iso_date.match(d) if m: return int(m.group(1)) m = re_end_year.match(d) if m: return int(m.group(1)) {'lc_classifications': ['PZ7.H558 Ru'], 'dewey_number': ['[E]']}
from __future__ import print_function import sys import codecs from openlibrary.catalog.utils.query import query_iter, set_staging, query from openlibrary.api import OpenLibrary, Reference from openlibrary.catalog.read_rc import read_rc from time import sleep set_staging(True) rc = read_rc() ol = OpenLibrary("http://dev.openlibrary.org") ol.login('EdwardBot', rc['EdwardBot']) sys.stdout = codecs.getwriter('utf-8')(sys.stdout) 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'])
from __future__ import print_function import sys import codecs from openlibrary.catalog.utils.query import query_iter, set_staging, query from openlibrary.api import OpenLibrary, Reference from openlibrary.catalog.read_rc import read_rc from time import sleep set_staging(True) rc = read_rc() ol = OpenLibrary("http://dev.openlibrary.org") ol.login('EdwardBot', rc['EdwardBot']) sys.stdout = codecs.getwriter('utf-8')(sys.stdout) 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']
from openlibrary.catalog.read_rc import read_rc from collections import defaultdict from pprint import pformat from openlibrary.catalog.utils.edit import fix_edition from openlibrary.catalog.importer.db_read import get_mc import urllib2 from openlibrary.api import OpenLibrary, Reference from lxml import etree from time import sleep, time import six rc = read_rc() ol = OpenLibrary("http://openlibrary.org") ol.login('WorkBot', rc['WorkBot']) def write_log(cat, key, title): print((("%.2f" % time()), cat, key, title), file=fh_log) fh_log.flush() sys.stdout = codecs.getwriter('utf-8')(sys.stdout) re_skip = re.compile( '\b([A-Z]|Co|Dr|Jr|Capt|Mr|Mrs|Ms|Prof|Rev|Revd|Hon|etc)\.$') re_ia_marc = re.compile('^(?:.*/)?([^/]+)_(marc\.xml|meta\.mrc)(:0:\d+)?$') ns = '{http://www.loc.gov/MARC21/slim}' ns_leader = ns + 'leader'
"--bot-username", dest="username", default="nyt_bestsellers_bot", help="The bot username for accessing the Open Library API") op.add_option("-p", "--bot-password", dest="password", help="The bot password for accessing the Open Library API") options, _ = op.parse_args() global NYT_API_KEY NYT_API_KEY = options.nyt_api_key global OL OL = OpenLibrary("http://%s" % options.openlibrary_host) OL.login(options.username, options.password) results = collections.defaultdict(list) for ln in get_nyt_bestseller_list_names(): LOG("INFO", "processing list %s" % ln) for i, book in enumerate(load_nyt_bestseller_list(ln)): ol_keys = reconcile_book(book) if not ol_keys: LOG( "WARN", "unable to reconcile '%s' by %s - no OL book found" % (book['book_details'][0]['title'], book['book_details'][0]['author'])) if not (key for key in ol_keys if key.startswith("/works/")): LOG( "WARN", "only editions for '%s' by %s: %s" % (book['book_details'][0]['title'],
import argparse parser = argparse.ArgumentParser(description='scribe loader') parser.add_argument('--skip_hide_books', action='store_true') parser.add_argument('--item_id') parser.add_argument('--config', default='openlibrary.yml') args = parser.parse_args() config_file = args.config config.load(config_file) import_bot_password = config.runtime_config['load_scribe']['import_bot_password'] # '/1/var/log/openlibrary/load_scribe' load_scribe_log = config.runtime_config['load_scribe']['log'] ol = OpenLibrary("http://openlibrary.org") ol.login('ImportBot', import_bot_password) password = Popen(["/opt/.petabox/dbserver"], stdout=PIPE).communicate()[0] db = web.database(dbn='mysql', host='dbmeta.us.archive.org', user='******', \ passwd=password, db='archive') db.printing = False re_census = re.compile('^\d+(st|nd|rd|th)census') re_edition_key = re.compile('^/(?:books|b)/(OL\d+M)$') def read_short_title(title): return str(fast_parse.normalize_str(title)[:25]) def make_index_fields(rec): fields = {}
from openlibrary.catalog.marc import fast_parse, is_display_marc from openlibrary.catalog.marc.parse import read_edition, NoTitle from openlibrary.catalog.marc.marc_subject import subjects_for_work from time import time, sleep from openlibrary.api import OpenLibrary, unmarshal from pprint import pprint import argparse parser = argparse.ArgumentParser(description="scribe loader") parser.add_argument("--skip_hide_books", action="store_true") parser.add_argument("--item_id") args = parser.parse_args() rc = read_rc() ol = OpenLibrary("http://openlibrary.org") ol.login("ImportBot", rc["ImportBot"]) db_amazon = web.database(dbn="postgres", db="amazon") db_amazon.printing = False db = web.database(dbn="mysql", host=rc["ia_db_host"], user=rc["ia_db_user"], passwd=rc["ia_db_pass"], db="archive") db.printing = False re_census = re.compile("^\d+(st|nd|rd|th)census") re_edition_key = re.compile("^/(?:books|b)/(OL\d+M)$") def read_short_title(title): return str(fast_parse.normalize_str(title)[:25])
import MySQLdb import re from openlibrary.api import OpenLibrary, Reference from collections import defaultdict import six re_edition_key = re.compile('^/books/OL(\d+)M$') 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]
from openlibrary.catalog.utils.query import query_iter, set_staging, query from openlibrary.catalog.utils import mk_norm from openlibrary.catalog.read_rc import read_rc from collections import defaultdict from pprint import pprint, pformat from openlibrary.catalog.utils.edit import fix_edition from openlibrary.catalog.importer.db_read import get_mc import urllib2 from openlibrary.api import OpenLibrary, Reference from lxml import etree from time import sleep, time rc = read_rc() ol = OpenLibrary("http://openlibrary.org") ol.login('WorkBot', rc['WorkBot']) fh_log = open('/1/edward/logs/WorkBot', 'a') def write_log(cat, key, title): print >> fh_log, (("%.2f" % time()), cat, key, title) sys.stdout = codecs.getwriter('utf-8')(sys.stdout) re_skip = re.compile('\b([A-Z]|Co|Dr|Jr|Capt|Mr|Mrs|Ms|Prof|Rev|Revd|Hon|etc)\.$') re_ia_marc = re.compile('^(?:.*/)?([^/]+)_(marc\.xml|meta\.mrc)(:0:\d+)?$') ns = '{http://www.loc.gov/MARC21/slim}' ns_leader = ns + 'leader' ns_data = ns + 'datafield' def has_dot(s):
from openlibrary.api import OpenLibrary from catalog.read_rc import read_rc import six rc = read_rc() marc_index = web.database(dbn='postgres', db='marc_index') marc_index.printing = False db_amazon = web.database(dbn='postgres', db='amazon') db_amazon.printing = False ol = OpenLibrary("http://openlibrary.org") ol.login('ImportBot', rc['ImportBot']) sys.stdout = codecs.getwriter('utf-8')(sys.stdout) t0 = time() t_prev = time() rec_no = 0 chunk = 50 load_count = 0 archive_id = sys.argv[1] def percent(a, b): return float(a * 100.0) / b def progress(archive_id, rec_no, start_pos, pos):
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)