import zipfile
from bs4 import BeautifulSoup
import os
import re

import extras
import config

class Entry(object):
    valid_keys = (
        "id",
        "url",
        "title",
        "content",
        "downloadsPerMonth",
        "updated",
        "identifier",
        "date",
        "rights",
        "summary",
        "dcterms_source",
        "provider",
        "publishers",
        "contributors",
        "languages",
        "subjects",
        "oai_updatedates",
        "authors",
        "formats",
        "size",
        "links",
        "cover",
        "covertype"
    )

    required_keys = ("id", "title", "links")

    def validate(self, key, value):
        if key not in Entry.valid_keys:
            raise KeyError("invalid key in opds.catalog.Entry: %s" % (key))

    def __init__(self, **kwargs):
        for key, val in kwargs.items():
            self.validate(key, val)

        for req_key in Entry.required_keys:
            if not req_key in kwargs:
                raise KeyError("required key %s not supplied for Entry!" % (req_key))
        self.id = kwargs["id"]
        self.title = kwargs["title"]
        self.links = kwargs["links"]
        self._data = kwargs
        
        #print(">>entry.py")
        #print(kwargs)
        #print(kwargs["links"][0].get("rpath"))
        #print("--end entry.py")

        if kwargs["links"][0].get("type") == 'application/x-cbz':
            f=self.links[0].get("rpath")+"/"+self.title+".cbz"
            if os.path.exists(f): 
                s = zipfile.ZipFile(f)
                self.size = extras.get_size(f, 'mb')
                data=BeautifulSoup(s.open('ComicInfo.xml').read(), "xml")
                #self.cover=s.open('P00001.jpg').read()
                self.authors = data.select('Writer')[0].text.split(",")
                self.cover = "/image/" + extras.get_cvdb(data.select('Notes')) + ".jpg"
                

            #print(data)
            #print(kwargs["links"][0])
            #print(data.select('Series')[0].text)
            #print(kwargs["links"][0].get("rpath"))
                if data.select('Series')[0].text in kwargs["links"][0].get("rpath"):
                    releasedate=data.select('Year')[0].text+"-"+data.select('Month')[0].text.zfill(2)+"-"+data.select('Day')[0].text.zfill(2)
                    try:
                        self.title = "#"+data.select('Number')[0].text.zfill(2) + ": " + data.select('Title')[0].text + " (" + releasedate + ") [" + str(self.size) + "MB]"
                    except:
                        self.title = "#"+data.select('Number')[0].text.zfill(2) + " (" + releasedate + ") [" + str(self.size) + "MB]"
            #print(self.title)
                else:
                    self.title = kwargs["title"]

            else:
                self.title = kwargs["title"]
            #self.title = data.select('Title')[0].text
    def get(self, key):
        return self._data.get(key, None)

    def set(self, key, value):
        self.validate(key, value)
        self._data[key] = value