ComicOPDS/opds/catalog.py

import os
from uuid import uuid4
from urllib.parse import quote
from jinja2 import Environment, FileSystemLoader, select_autoescape
from .entry import Entry
from .link import Link
import sqlite3,json
import config

class Catalog(object):
    def __init__(
        self,
        title,
        id=None,
        author_name=None,
        author_uri=None,
        root_url=None,
        url=None,
    ):
        self.title = title
        self.id = id or uuid4()
        self.author_name = author_name
        self.author_uri = author_uri
        self.root_url = root_url
        self.url = url
        self.entries = []

    def add_entry(self, entry):
        self.entries.append(entry)

    def render(self):
        env = Environment(
            loader=FileSystemLoader(
                searchpath=os.path.join(os.path.dirname(__file__), "templates")
            ),
            autoescape=select_autoescape(["html", "xml"]),
        )
        template = env.get_template("catalog.opds.jinja2")
        return template.render(catalog=self)

def fromsearch(root_url, url, content_base_path, content_relative_path):

    c = Catalog(
        title="test"
    )
        
    return c

def fromdir(root_url, url, content_base_path, content_relative_path):


    path = os.path.join(content_base_path, content_relative_path)

    if os.path.basename(content_relative_path) == "":
        c = Catalog(
            title="Comics",
            root_url=root_url,
            url=url
            )
    else:
        c = Catalog(
            title=os.path.basename(content_relative_path), 
            root_url=root_url, 
            url=url
        )
    #title=os.path.basename(os.path.dirname(path)), root_url=root_url, url=url

    ##########WORKING AREA###########
    searchArr=[]
    if c.url.endswith("/catalog"):
        with open('test.json') as fi:
            data=json.load(fi)
            print("--> LOADED FILE") # try and get this as low as possible.
    #searchArr=["Girl","Bat","Part One"]
    
            for e in data:
                for key, value in e.items():
                    searchArr.append(key)
    print(searchArr) 
    ######################

    
    

    if not "search" in c.url:
        onlydirs = [
            f for f in os.listdir(path) if not os.path.isfile(os.path.join(path, f))
        ]
    #print(onlydirs)
        for dirname in onlydirs:
            link = Link(
                href=quote(f"/catalog/{content_relative_path}/{dirname}").replace('//','/'), #windows fix
                rel="subsection",
                rpath=path,
                type="application/atom+xml;profile=opds-catalog;kind=acquisition",
            )
            c.add_entry(Entry(title=dirname, id=uuid4(), links=[link]))

    
    if c.url.endswith("/catalog"):
        
        for i in searchArr:

            link2 = Link(
                href=quote(f"/catalog/search["+i+"]"),
                rel="subsection",
                rpath=path,
                type="application/atom+xml;profile=opds-catalog;kind=acquisition",
            )
            c.add_entry(Entry(title="["+i+"]",id=uuid4(),links=[link2]))

    if not "search" in c.url:
        onlyfiles = [f for f in os.listdir(path) if os.path.isfile(os.path.join(path, f))]
        onlyfiles.sort()
        for filename in onlyfiles:
            link = Link(
                href=quote(f"/content/{content_relative_path}/{filename}"),
                rel="http://opds-spec.org/acquisition",
                rpath=path,
                type=mimetype(filename),
            )
            c.add_entry(Entry(title=filename.rsplit(".",1)[0], id=uuid4(), links=[link]))
            #fixed issue with multiple . in filename
            #print(c.render()) 
    else:
        with open('test.json') as fi:
            data=json.load(fi)
            print("--> LOADED 2 FILE") # try and get this as low as possible.
        for e in data:
            for key, value in e.items():
                #print(key)
                searchArr.append(key)
        for i in searchArr:
            #print(i)
            if quote(f""+i) in c.url:
                conn = sqlite3.connect('app.db')
                #print(data)
                for e in data:
                    for key, value in e.items():
                        print(key)
                        if key == i:
                            query="SELECT * FROM COMICS where "
                            for i in value:
                                first=True
                                for j,k in i.items():
                                    if j == 'SQL':
                                        query = query + k
                                    if k != '' and j != "SQL":
                                        print(j,k)
                                        if not first:
                                            query = query + "and "
                                        if type(k) == list:
                                            print(k)
                                            if j == "series" or j == "title":
                                                firstS = True
                                                query = query + "("
                                                for l in k:
                                                    if not firstS:
                                                        query = query + "or "
                                                    query = query + j + " like '%" + l + "%' "
                                                    if firstS: 
                                                        firstS = False
                                                query = query + ") "
                                            else:
                                                query = query + j + " in (" 
                                                firstL = True
                                                for l in k:
                                                    if not firstL: 
                                                        query = query + ","
                                                    query = query + "'" + l + "'"
                                                    if firstL:
                                                        firstL = False
                                                query = query + ") "

                                        else:
                                            query = query + j + " like '%" + k + "%' "
                                        if first:
                                            first = False
                            query = query + " order by series asc, cast(issue as unsigned) asc "
                            if config.DEFAULT_SEARCH_NUMBER != 0:
                                query = query + "LIMIT " + str(config.DEFAULT_SEARCH_NUMBER) + ";"
                            else:
                                query = query + ";"
                print("----> " + query)
                                
                sql = query
                #sql="SELECT * from COMICS where SERIES like '%" + i+ "%' or Title like '%" + i+ "%';"
                #print(sql)
                s = conn.execute(sql)
                #list=[] 
                for r in s:
                    #print(r)
                    tUrl=f""+r[7].replace('\\','/').replace(config.WIN_DRIVE_LETTER + ':','').replace(config.CONTENT_BASE_DIR,"/content")
                    print(tUrl)
                    tTitle=r[6]
                    link3 = Link(
                        #href=quote(f"/content/DC Comics/Earth Cities/Gotham City/Batgirl/Annual/(2012) Batgirl Annual/Batgirl Annual #001 - The Blood That Moves Us [December, 2012].cbz"),
                        href=quote(tUrl),
                        rel="http://opds-spec.org/acquisition",
                        rpath=path,
                        type="application/x-cbz",
                    )
                    print(link3.href)
                    c.add_entry(
                        Entry(
                            title=tTitle,
                            id=uuid4(),
                            links=[link3]
                            )
                        )
    #print(c.title)
    return c




def mimetype(path):
    extension = path.split(".")[-1].lower()
    if extension == "pdf":
        return "application/pdf"
    elif extension == "epub":
        return "application/epub"
    elif extension == "mobi":
        return "application/mobi"
    elif extension == "cbz":
        return "application/x-cbz"
    else:
        return "application/unknown"