myx_mam.py

import requests
import json
import os
import pickle
from pprint import pprint
import myx_classes
import myx_utilities


#MAM Functions
def searchMAM(cfg, titleFilename, authors, extension):
    #Config
    session = cfg.get("Config/session")
    log_path = cfg.get("Config/log_path")
    ebook = bool(cfg.get("Config/flags/ebooks"))
    audiobook = not (ebook)
    
    #put paren around authors and titleFilename
    if len(authors):
        authors = f"({authors})"

    if len(titleFilename):
        titleFilename = f"({titleFilename})"

    search = f'{authors} {titleFilename} {extension} @dummy mamDummy'

    #cache results for this search string
    cacheKey=myx_utilities.getHash(search)
    
    if myx_utilities.isCached(cacheKey, "mam", cfg):
        #this search has been done before, load results from cache
        results = myx_utilities.loadFromCache(cacheKey, "mam")
        return (results["data"])
    
    else:
        #save cookie for future use
        cookies_filepath = os.path.join(log_path, 'cookies.pkl')
        sess = requests.Session()

        #a cookie file exists, use that
        if os.path.exists(cookies_filepath):
            cookies = pickle.load(open(cookies_filepath, 'rb'))
            sess.cookies = cookies
        else:
            #assume a session ID is passed as a parameter
            sess.headers.update({"cookie": f"mam_id={session}"})

        #test session and cookie
        r = sess.get('https://www.myanonamouse.net/jsonLoad.php', timeout=5)  # test cookie
        if r.status_code != 200:
            raise Exception(f'Error communicating with API. status code {r.status_code} {r.text}')
        else:
            # save cookies for later
            with open(cookies_filepath, 'wb') as f:
                pickle.dump(sess.cookies, f)

            mam_categories = []
            if audiobook:
                mam_categories.append(13) #audiobooks
                mam_categories.append(16) #radio
            if ebook:
                mam_categories.append(14)
            if not mam_categories:
                return None
            
            params = {
                "tor": {
                    "text": search,  # The search string.
                    "srchIn": {
                        "title": "true",
                        "author": "true",
                        "fileTypes": "true",
                        "filenames": "true"
                    },
                    "main_cat": mam_categories
                },
                "perpage":50
            }

            try:
                r = sess.post('https://www.myanonamouse.net/tor/js/loadSearchJSONbasic.php', json=params)
                if r.text == '{"error":"Nothing returned, out of 0"}':
                    return None

                results = r.json()

                #cache this result before returning it
                myx_utilities.cacheMe(cacheKey, "mam", results, cfg)

                return (results["data"])
        
            except Exception as e:
                print(f'error searching MAM {e}')

    return None

def getMAMBook(cfg, titleFilename="", authors="", extension=""):
    books=[]
    mamBook=searchMAM(cfg, titleFilename, authors, extension)
    if (mamBook is not None):
        for b in mamBook:
            #pprint(b)
            book=myx_classes.Book()
            book.init()
            if 'asin' in b: 
                book.asin=str(b["asin"])
            if 'title' in b: 
                book.title=str(b["title"])
            if 'author_info'in b:
                #format {id:author, id:author}
                if len(b["author_info"]):
                    authors = json.loads(b["author_info"])
                    for author in authors.values():
                        book.authors.append(myx_classes.Contributor(str(author)))
            if 'series_info'in b:
                #format {"35598": ["Kat Dubois", "5"]}
                if len(b["series_info"]):
                    series_info = json.loads(b["series_info"])
                    for series in series_info.values():
                        s=list(series)
                        book.series.append(myx_classes.Series(str(s[0]), s[1]))    
            if 'lang_code' in b: 
                book.language=myx_utilities.getLanguage((b["lang_code"]))
            if 'my_snatched' in b:
                book.snatched=bool((b["my_snatched"])) 
            
            if book.snatched:
                books.append(book)

    return books