mirror of
https://github.com/kovidgoyal/calibre.git
synced 2025-06-23 15:30:45 -04:00
92 lines
4.1 KiB
Python
92 lines
4.1 KiB
Python
## Copyright (C) 2006 Kovid Goyal kovid@kovidgoyal.net
|
|
## This program is free software; you can redistribute it and/or modify
|
|
## it under the terms of the GNU General Public License as published by
|
|
## the Free Software Foundation; either version 2 of the License, or
|
|
## (at your option) any later version.
|
|
##
|
|
## This program is distributed in the hope that it will be useful,
|
|
## but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
## GNU General Public License for more details.
|
|
##
|
|
## You should have received a copy of the GNU General Public License along
|
|
## with this program; if not, write to the Free Software Foundation, Inc.,
|
|
## 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
import sqlite3 as sqlite
|
|
import os, os.path, zlib
|
|
from stat import ST_SIZE
|
|
from libprs500.lrf.meta import LRFMetaFile
|
|
|
|
|
|
class LibraryDatabase(object):
|
|
|
|
BOOKS_SQL = """
|
|
create table if not exists books_meta(id INTEGER PRIMARY KEY, title TEXT, authors TEXT, publisher TEXT, size INTEGER, tags TEXT,
|
|
cover BLOB, date DATE DEFAULT CURRENT_TIMESTAMP );
|
|
create table if not exists books_data(id INTEGER, extension TEXT, data BLOB);
|
|
"""
|
|
|
|
def __init__(self, dbpath):
|
|
self.con = sqlite.connect(dbpath)
|
|
self.con.row_factory = sqlite.Row # Allow case insensitive field access by name
|
|
self.con.executescript(LibraryDatabase.BOOKS_SQL)
|
|
|
|
def get_cover(self, id):
|
|
raw = self.con.execute("select cover from books_meta where id=?", (id,)).next()["cover"]
|
|
return zlib.decompress(str(raw)) if raw else None
|
|
|
|
def get_extensions(self, id):
|
|
exts = []
|
|
cur = self.con.execute("select extension from books_data where id=?", (id,))
|
|
for row in cur:
|
|
exts.append(row["extension"])
|
|
return exts
|
|
|
|
def add_book(self, path):
|
|
file = os.path.abspath(path)
|
|
title, author, publisher, size, cover = os.path.basename(file), None, None, os.stat(file)[ST_SIZE], None
|
|
ext = title[title.rfind(".")+1:].lower() if title.find(".") > -1 else None
|
|
if ext == "lrf":
|
|
lrf = LRFMetaFile(open(file, "r+b"))
|
|
title, author, cover, publisher = lrf.title, lrf.author.strip(), lrf.thumbnail, lrf.publisher.strip()
|
|
if "unknown" in publisher.lower(): publisher = None
|
|
if "unknown" in author.lower(): author = None
|
|
file = zlib.compress(open(file).read())
|
|
if cover: cover = sqlite.Binary(zlib.compress(cover))
|
|
self.con.execute("insert into books_meta (title, authors, publisher, size, tags, cover) values (?,?,?,?,?,?)", (title, author, publisher, size, None, cover))
|
|
id = self.con.execute("select max(id) from books_meta").next()[0]
|
|
self.con.execute("insert into books_data values (?,?,?)", (id, ext, sqlite.Binary(file)))
|
|
self.con.commit()
|
|
|
|
def get_table(self, columns):
|
|
cols = ",".join([ c for c in columns])
|
|
cur = self.con.execute("select " + cols + " from books_meta")
|
|
rows = []
|
|
for row in cur:
|
|
r = {}
|
|
for c in columns: r[c] = row[c]
|
|
rows.append(r)
|
|
return rows
|
|
|
|
def get_meta_data(self, id):
|
|
try: row = self.con.execute("select * from books_meta where id=?", (id,)).next()
|
|
except StopIteration: return None
|
|
data = {}
|
|
for field in ("id", "title", "authors", "publisher", "size", "tags", "cover", "date"):
|
|
data[field] = row[field]
|
|
return data
|
|
|
|
|
|
def search(self, query): pass
|
|
|
|
if __name__ == "__main__":
|
|
lbm = LibraryDatabase("/home/kovid/library.sqlite")
|
|
# lbm.add_book("/home/kovid/documents/ebooks/hobbfar01.lrf")
|
|
# lbm.add_book("/home/kovid/documents/ebooks/hobbfar02.lrf")
|
|
# lbm.add_book("/home/kovid/documents/ebooks/hobbfar03.lrf")
|
|
# lbm.add_book("/home/kovid/documents/ebooks/hobblive01.lrf")
|
|
# lbm.add_book("/home/kovid/documents/ebooks/hobbtawny01.lrf")
|
|
# lbm.add_book("/home/kovid/documents/ebooks/hobbtawny02.lrf")
|
|
# lbm.add_book("/home/kovid/documents/ebooks/hobbtawny03.lrf")
|
|
print lbm.get_table(["id","title"])
|