fic-archive/ships.py

import datetime, os
from importlib import import_module

fffandoms = ["FF1","FF2","FF3","FF4","FF5","FF6","FF7","FF8","FF9","FFX","FF11","FF12","FF13","FF14","FF15","FF16"]

import makeheader
import headerfooter

def shiplist(local=False):
    # delete existing file
    if os.path.exists("build/ff/ships/index.html"):
        os.remove("build/ff/ships/index.html")
    # write header
    headerfooter.headerwrite("build/ff/ships/index.html","FF fics by ship","FF fics by ship","<p>Click on each bar to see fics involving that ship, organised according to whether it’s the main ship or a secondary one and then from newest to oldest.</p>",False,local)
    # iterate through fandoms
    ships = []
    numbers = ["one","two","three","four","five","six","seven","eight","nine","ten","eleven","twelve","thirteen","fourteen","fifteen","sixteen"]
    for fandom in fffandoms:
        theships = []
        ficcount = 500
        while ficcount > 0:
            ficcount -= 1
            if ficcount < 10:
                ficcountstring = "00" + str(ficcount)
            elif ficcount < 100:
                ficcountstring = "0" + str(ficcount)
            else:
                ficcountstring = str(ficcount)
            if os.path.exists("originalsmeta/" + ficcountstring + ".py"):
                ficfile = "originalsmeta." + ficcountstring
                fileread = import_module(ficfile)
                try:
                    if fileread.revealdate > datetime.datetime.now():
                        revealed = False
                    else:
                        revealed = True
                except:
                    revealed = True
                if revealed == True:
                    if len(fileread.fandom) == 1:
                        if fandom in fileread.fandom:
                            try:
                                theships.extend(fileread.ship)
                            except:
                                pass
        for ship in theships:
            if ship == None:
                theships.remove(ship)
        theships = sorted(list(dict.fromkeys(theships)))
        for ship in theships:
            shipdict = {"pairing":ship,"game":(fffandoms.index(fandom) + 1)}
            ships.append(shipdict)
    shiplist = sorted(ships, key=lambda d: d["pairing"])
    for aship in shiplist:
        maincount = []
        secondarycount = []
        ship = aship["pairing"]
        searchfandom = fffandoms[(aship["game"]) - 1]
        cssgame = numbers[(aship["game"]) - 1]
        # check all fics in specified fandom
        ficcount = 500
        while ficcount > 0:
            ficcount -= 1
            if ficcount < 10:
                ficcountstring = "00" + str(ficcount)
            elif ficcount < 100:
                ficcountstring = "0" + str(ficcount)
            else:
                ficcountstring = str(ficcount)
            if os.path.exists("originalsmeta/" + ficcountstring + ".py"):
                countfile = "originalsmeta." + ficcountstring
                fileread = import_module(countfile)
                try:
                    if fileread.revealdate > datetime.datetime.now():
                        revealed = False
                    else:
                        revealed = True
                except:
                    revealed = True
                if revealed == True:
                    ficdict = {"thefic":ficcount,"thedate":(fileread.datewords[-1])["date"]}
                    if searchfandom in fileread.fandom:
                        # append to lists
                        try:
                            if fileread.ship[0] == ship:
                                maincount.append(ficdict)
                            elif ship in fileread.ship:
                                secondarycount.append(ficdict)
                        except:
                            pass
        # write details element
        output = "build/ff/ships/index.html"
        filewrite = open(output, "a")
        filewrite.write("<details><summary><span class=\"character " + cssgame + "\">" + ship.replace("/"," × ") + "</span> ")
        # write statistics
        if len(maincount) > 0:
            filewrite.write("<span class=\"main\"> main: " + str(len(maincount)) + "</span>")
        if len(secondarycount) > 0:
            filewrite.write("<span class=\"secondary\"> secondary: " + str(len(secondarycount)) + "</span>")
        filewrite.write("</summary>\n")
        filewrite.close()
        # write fic headers in each category
        if len(maincount) > 0:
            filewrite = open(output, "a")
            filewrite.write("<h1>Main ship</h1>\n")
            filewrite.close()
            maincount = sorted(maincount,key=lambda d: d["thedate"],reverse=True)
            newmain = []
            for fic in maincount:
                newmain.append(fic["thefic"])
            for fic in newmain:
                makeheader.ficgen(fic,False,output,local)
        if len(secondarycount) > 0:
            filewrite = open(output, "a")
            filewrite.write("<h1>Secondary ship</h1>\n")
            filewrite.close()
            secondarycount = sorted(secondarycount,key=lambda d: d["thedate"],reverse=True)
            newsecondary = []
            for fic in secondarycount:
                newsecondary.append(fic["thefic"])
            for fic in newsecondary:
                makeheader.ficgen(fic,False,output,local)
        filewrite = open(output, "a")
        filewrite.write("</details>\n")
        filewrite.close()
    # write footer
    headerfooter.footerwrite("build/ff/ships/index.html",False,local)

if __name__ == "__main__":
    shiplist()