webgen/webgen.py

#!/usr/bin/env python3

import sys, os, uuid, shutil, subprocess, markdown, re, datetime, locale, glob
from dialog import Dialog


# If not choosen stdout as output, where should the generated file go to and what
# fileending (f.e. .html) shall it have
targetdir = os.path.expanduser("~/www/i21k.de/posts/")
sourcedir = os.path.expanduser("~/python/webgen/")
fileending = ""
locale.setlocale(locale.LC_ALL, "")
clear = "\x1b[2J\x1b[H"
xwidth = os.get_terminal_size()[0] - 3
ywidth = os.get_terminal_size()[1] - 3
os.chdir(sourcedir)

# Prepend an hourly tinestamp to the later name of the file
creationtime = datetime.datetime.now().strftime("%Y-%m-%d %H:00 ")
# generate a more precise ctime to be included in the resulting webpage
creationtimeheader = datetime.datetime.now().strftime("%Y-%m-%d %H:%M")


# Initialize a dialog.Dialog instance
d = Dialog(dialog="dialog")

# call favorite editor with filename to write the text
def edit(headline):
    editor = os.getenv("EDITOR")
    if not editor:
        editor = "vim"
    subprocess.call([editor, headline])
    return headline


# Select a topic for the new post or reedit an existing one
def selectfile():
    # Put files in sourcefolder into list
    zahl = 0
    tabelle = []
    for zeile in glob.glob("*.md"):
        tabelle.insert(zahl, (str(zahl), zeile))
        zahl += 1

    # Open list in menu and let user choose one
    ausgewaehlt = d.menu(
        "Such dir ne Datei:",
        width=xwidth,
        height=ywidth,
        menu_height=ywidth,
        title="Ein Thema wählen:",
        choices=tabelle,
        cancel="Neuen Namen eingeben",
    )

    if ausgewaehlt[0] == "cancel":
        newtopic = d.inputbox("Ok, gib hier ein neues Thema an:", width=xwidth, height=ywidth, title="Ein Thema wählen:", cancel="Exit")
        if newtopic[0] == "cancel" or newtopic[1] == "":
            sys.exit("Na, dann eben nicht...")
        headline = newtopic[1]
    else:
        headline = ""

    # Clean filename from leading .md for later construction of a headline
    if headline == "":
        headline = re.sub("\.md$", "", tabelle[int(ausgewaehlt[1])][1])

    # if the searched filename doesnt match an existing file we construct a new one
    if len(glob.glob(sourcedir + headline + ".md")) == 0:
        sourcefile = sourcedir + headline + ".md"
    else:
        sourcefile = str(glob.glob(sourcedir + headline + ".md")[0])

    # if the targetfilename doesnt match an existing file we construct a new one
    targetfile = glob.glob(targetdir + "[0-9][0-9][0-9][0-9]-[0-9][0-9]-[0-9][0-9] [0-2][0-9]:[0-6][0-9]        " + headline + "*")
    targetfilelength = len(targetfile)
    if targetfilelength >= 1:
        targetfile = targetfile[targetfilelength - 1]
    else:
        targetfile = targetdir + datetime.datetime.now().strftime("%Y-%m-%d %H:00        ") + headline

    return sourcefile, targetfile, headline


# Parse arguments "--html", "--stdout" and "Some Headline"
# there are 3 parameters. if --html is given, the fileending will be .html.
# Else, if stdout is choosen there can be no fileending, but it will still contain html
# the name of the program is stored in myname
# dont set a name to outfile, it just is initialised here and will be overwritten

optwords = ["--html", "--stdout"]
outfile = ""
args = sys.argv
myname = args.pop(0)
for wish in optwords:
    if args.count(wish):
        if wish == "--html":
            fileending = ".html"
            args.pop(args.index(wish))
        if wish == "--stdout":
            outfile = sys.stdout
            args.pop(args.index(wish))
# Check, if that topic already exists and fetch the appropriate filename
if len(args) == 1:
    headline = args[0]
    inputfile = headline + ".md"
    outputfile = glob.glob(targetdir + "[0-9][0-9][0-9][0-9]-[0-9][0-9]-[0-9][0-9] [0-2][0-9]:[0-6][0-9]        " + headline + "*")
    outputfilelength = len(outputfile)
    if outputfilelength >= 1:
        outputfile = outputfile[outputfilelength - 1]
    else:
        outputfile = targetdir + datetime.datetime.now().strftime("%Y-%m-%d %H:00        ") + headline


elif len(args) == 0:
    selection = selectfile()
    inputfile = selection[0]
    outputfile = selection[1]
    headline = selection[2]
elif len(args) > 1:
    sys.exit("Too much or wrong parameters:\n" + myname + ' [--html] [--stdout] ["Some Topic"] ')


# This is informal asking yes/no for editing the text
if outfile == sys.stdout:
    outputfile = "STDOUT"
print("\nYou will edit now: '" + inputfile + "'\nOutputfile= '" + outputfile + "'")
a = input("\nPress <RETURN> to continue or <CTRL-C> to stop: ")
edit(inputfile)


# Set ouput to stdout, if requested. Caution, the prompts of this script go there, too,
# be careful with copy & paste
if outfile != sys.stdout:
    html_out_file = open(outputfile, "w")
else:
    html_out_file = outfile

# The head of the output. Strip path from myname
head1 = (
    '<!DOCTYPE HTML><html><head>\n\
<meta http-equiv="content-type" content="text/html; charset=UTF-8">\n<title>'
    + headline
    + '</title>\n\
<meta name="syntax" content="markdown">\n\
<meta name="generator" content="'
    + myname.rpartition("/")[2]
    + ' markdown2htmlconverter">\n\
<meta name="settings" content="use_css,pre_wrap,no_foldcolumn,expand_tabs,prevent_copy=">\n\
<meta name="ctime" content="'
    + creationtimeheader
    + '"> '
)

# Some fancy css for minimalistic terminal style

style_fn = '<link rel="stylesheet" type="text/css" href="/vimstyles.css">'
head2 = "</head><body>\n"
body1 = '<a href="./">Back</a>'
foot = "</body> </html>\n"

# write beginning of html-file
html_out_file.write(head1 + style_fn + head2 + body1)

# The centerpiece - read md-file, convert to html, add head and foot and write result
with open(inputfile, "r", encoding="utf-8") as infile:
    md_data = infile.read()
    html_output = markdown.markdown(md_data)
    html_out_file.write(html_output)
    html_out_file.write(foot)
html_out_file.close()

# if the new filename is not stdout, return new filename to stdout, so some calling shellscript can use it.
if outfile != sys.stdout:
    print(outputfile)

# Have a nice time.