2016-06-18 06:50:32 -07:00
|
|
|
|
|
|
|
from lxml import etree
|
|
|
|
import os
|
|
|
|
import uploader
|
|
|
|
|
2017-01-28 11:16:40 -08:00
|
|
|
# ToDo: Check usage of original_file_name
|
2016-06-18 06:50:32 -07:00
|
|
|
def get_fb2_info(tmp_file_path, original_file_name, original_file_extension):
|
|
|
|
|
|
|
|
ns = {
|
2016-12-23 00:53:39 -08:00
|
|
|
'fb': 'http://www.gribuser.ru/xml/fictionbook/2.0',
|
|
|
|
'l': 'http://www.w3.org/1999/xlink',
|
2016-06-18 06:50:32 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
fb2_file = open(tmp_file_path)
|
|
|
|
tree = etree.fromstring(fb2_file.read())
|
|
|
|
|
|
|
|
authors = tree.xpath('/fb:FictionBook/fb:description/fb:title-info/fb:author', namespaces=ns)
|
2016-12-23 00:53:39 -08:00
|
|
|
|
2016-06-18 06:50:32 -07:00
|
|
|
def get_author(element):
|
2016-12-23 00:53:39 -08:00
|
|
|
return element.xpath('fb:first-name/text()', namespaces=ns)[0] + ' ' + element.xpath('fb:middle-name/text()',
|
|
|
|
namespaces=ns)[0] + ' ' + element.xpath('fb:last-name/text()', namespaces=ns)[0]
|
2016-06-18 06:50:32 -07:00
|
|
|
author = ", ".join(map(get_author, authors))
|
|
|
|
|
|
|
|
title = unicode(tree.xpath('/fb:FictionBook/fb:description/fb:title-info/fb:book-title/text()', namespaces=ns)[0])
|
2016-12-23 00:53:39 -08:00
|
|
|
description = unicode(tree.xpath('/fb:FictionBook/fb:description/fb:publish-info/fb:book-name/text()',
|
|
|
|
namespaces=ns)[0])
|
2016-06-18 06:50:32 -07:00
|
|
|
|
|
|
|
return uploader.BookMeta(
|
2016-12-23 00:53:39 -08:00
|
|
|
file_path=tmp_file_path,
|
|
|
|
extension=original_file_extension,
|
|
|
|
title=title,
|
|
|
|
author=author,
|
|
|
|
cover=None,
|
|
|
|
description=description,
|
|
|
|
tags="",
|
|
|
|
series="",
|
2016-06-18 06:50:32 -07:00
|
|
|
series_id="")
|