From 8b5bd61467d769cd7d1c0a9aab98045f0c275fba Mon Sep 17 00:00:00 2001
From: Pavel Yakunin <pyakunin@gmail.com>
Date: Sat, 18 Jun 2016 16:50:32 +0300
Subject: [PATCH] fb2 uploading

---
 cps/book_formats.py | 13 +++++++--
 cps/fb2.py          | 64 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 75 insertions(+), 2 deletions(-)
 create mode 100644 cps/fb2.py

diff --git a/cps/book_formats.py b/cps/book_formats.py
index 4e797113..b5c48086 100644
--- a/cps/book_formats.py
+++ b/cps/book_formats.py
@@ -22,16 +22,25 @@ try:
     import epub
     use_epub_meta = True
 except ImportError, e:
-    logger.warning('cannot import PyPDF2, extracting pdf metadata will not work: %s', e)
+    logger.warning('cannot import PyPDF2, extracting epub metadata will not work: %s', e)
     use_epub_meta = False
 
+try:
+    import fb2
+    use_fb2_meta = True
+except ImportError, e:
+    logger.warning('cannot import lxml, extracting fb2 metadata will not work: %s', e)
+    use_fb2_meta = False
+
 
 def process(tmp_file_path, original_file_name, original_file_extension):
     try:
         if ".PDF" == original_file_extension.upper():
             return pdf_meta(tmp_file_path, original_file_name, original_file_extension)
-        if ".EPUB" == original_file_extension.upper() and use_pdf_meta == True:
+        if ".EPUB" == original_file_extension.upper() and use_epub_meta == True:
             return epub.get_epub_info(tmp_file_path, original_file_name, original_file_extension)
+        if ".FB2" == original_file_extension.upper() and use_fb2_meta == True:
+            return fb2.get_fb2_info(tmp_file_path, original_file_name, original_file_extension)
     except Exception, e:
         logger.warning('cannot parse metadata, using default: %s', e)
 
diff --git a/cps/fb2.py b/cps/fb2.py
new file mode 100644
index 00000000..746d041e
--- /dev/null
+++ b/cps/fb2.py
@@ -0,0 +1,64 @@
+
+from lxml import etree
+import os
+import uploader
+
+
+def get_fb2_info(tmp_file_path, original_file_name, original_file_extension):
+
+    ns = {
+        'fb':'http://www.gribuser.ru/xml/fictionbook/2.0',
+        'l':'ttp://www.w3.org/1999/xlink',
+    }
+
+    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)
+    def get_author(element):
+        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]
+    author = ", ".join(map(get_author, authors))
+
+    title = unicode(tree.xpath('/fb:FictionBook/fb:description/fb:title-info/fb:book-title/text()', namespaces=ns)[0])
+    description = unicode(tree.xpath('/fb:FictionBook/fb:description/fb:publish-info/fb:book-name/text()', namespaces=ns)[0])
+
+    #
+    #
+    #
+    # cfname = tree.xpath('n:rootfiles/n:rootfile/@full-path',namespaces=ns)[0]
+    #
+    # cf = zip.read(cfname)
+    # tree = etree.fromstring(cf)
+    #
+    # p = tree.xpath('/pkg:package/pkg:metadata',namespaces=ns)[0]
+    #
+    # epub_metadata = {}
+    # for s in ['title', 'description', 'creator']:
+    #     tmp = p.xpath('dc:%s/text()'%(s),namespaces=ns)
+    #     if (len(tmp) > 0):
+    #         epub_metadata[s] = p.xpath('dc:%s/text()'%(s),namespaces=ns)[0]
+    #     else:
+    #         epub_metadata[s] = "Unknown"
+    #
+    # coversection = tree.xpath("/pkg:package/pkg:manifest/pkg:item[@id='cover']/@href",namespaces=ns)
+    # if (len(coversection) > 0):
+    #     coverfile = extractCover(zip, coversection[0], tmp_file_path)
+    # else:
+    #     coverfile = None
+    # if epub_metadata['title'] is None:
+    #     title = original_file_name
+    # else:
+    #     title = epub_metadata['title']
+    #
+    #
+    return uploader.BookMeta(
+        file_path = tmp_file_path,
+        extension = original_file_extension,
+        title = title,
+        author = author,
+        cover = None,
+        description = description,
+        tags = "",
+        series = "",
+        series_id="")
+