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
)
