def save_rdf_in_database(parser): # Insert author, if it not exists if parser.author_id: try: author_record = Author.get(gut_id=parser.author_id) if parser.last_name: author_record.last_name if parser.first_name: author_record.first_names = parser.first_name if parser.birth_year: author_record.birth_year = parser.birth_year if parser.death_year: author_record.death_year = parser.death_year author_record.save() except: author_record = Author.create( gut_id=parser.author_id, last_name=parser.last_name, first_names=parser.first_name, birth_year=parser.birth_year, death_year=parser.death_year) else: # No author, set Anonymous author_record = Author.get(gut_id='216') # Get license try: license_record = License.get(name=parser.license) except: license_record = None # Insert book book_record = Book.create( id=parser.gid, title=parser.title.strip(), subtitle=parser.subtitle.strip(), author=author_record, # foreign key license=license_record, # foreign key language=parser.language.strip(), downloads=parser.downloads ) # Insert formats for file_type in parser.file_types: # Sanitize MIME mime = parser.file_types[file_type] if not mime.startswith('text/plain'): mime = re.sub(r'; charset=[a-z0-9-]+', '', mime) # else: # charset = re.match(r'; charset=([a-z0-9-]+)', mime).groups()[0] # Insert format type pattern = re.sub(r'' + parser.gid, '{id}', file_type) pattern = pattern.split('/')[-1] bid = int(book_record.id) if bid in BAD_BOOKS_FORMATS.keys() \ and mime in [FORMAT_MATRIX.get(f) for f in BAD_BOOKS_FORMATS.get(bid)]: logger.error("\t**** EXCLUDING **** {} for book #{} from list." .format(mime, bid)) continue format_record = Format.get_or_create( mime=mime, images=file_type.endswith( '.images') or parser.file_types[file_type] == 'application/pdf', pattern=pattern) # Insert book format BookFormat.create( book=book_record, # foreign key format=format_record # foreign key )