#!/usr/bin/env python3
import sys, os, uuid, shutil, signal, 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
webbasedir = os.path.expanduser("~/www/i21k.de/")
targetdir = webbasedir + "posts/"
sourcedir = os.path.expanduser("~/python/webgen/")
### for module mkdirindex:
filesdir = webbasedir + "files/"
indexfilename = "index.html"
filesdirheadline = "Files"
####
fileending = ""
locale.setlocale(locale.LC_ALL, "")
clear = "\x1b[2J\x1b[H"
green = "\x1b[38;5;46m"
red = "\x1b[38;5;9m"
yellow = "\x1b[38;5;226m"
greenonblack = "\x1b[38;5;46m\x1b[48;5;16m"
redonblack = "\x1b[38;5;9m\x1b[48;5;16m"
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")
# Let's stop everything at CTRL-c
def sigint_handler(signum, frame):
sys.exit(yellow + "\nPfff....I'll tell your mom!\n")
signal.signal(signal.SIGINT, sigint_handler)
# 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 all .md files in sourcefolder into list with a format, which dialog can eat.
filelist = glob.glob("*.md")
filelist.sort()
tabelle = []
zahl = 0
for zeile in filelist:
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",
)
# Nothing has been choosen, so lets ask for a topic to generate a filename
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(yellow + "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])
print(headline)
a = input(yellow + "What do we dooo here?")
# 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
def mkdirindex(filesdir, indexfilename, filesdirheadline):
"""Dig the files-directory and generate an index."""
head1 = (
'\n \n'
+ filesdirheadline
+ '\n \n \n '
)
style_fn = ''
head2 = ""
body1 = 'Back
'
body1 += '
Some files. Maybe useful or not.
'
body1 += '
'
foot = "
You can't avoid chaos. \n"
linecounter=0
dirindexfile = open(filesdir + indexfilename, "w")
dirindexfile.write(head1 + style_fn + head2 + body1)
for file in os.listdir(filesdir):
if not re.match("^\.", file) and not file == indexfilename:
size = os.lstat(filesdir + file).st_size
if size > 100000000000:
sizestr = str(int(size / 1073741824)) + " GiB"
elif size > 100000000:
sizestr = str(int(size / 1048576)) + " MiB"
elif size > 10000:
sizestr = str(int(size / 1024)) + " KiB"
else:
sizestr = str(size) + " B"
line = str('
")
dirindexfile.write(line)
linecounter+=1
dirindexfile.write(foot)
return linecounter
# 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 + "*" + fileending
)
outputfilelength = len(outputfile)
if outputfilelength >= 1:
outputfile = outputfile[outputfilelength - 1]
else:
outputfile = targetdir + datetime.datetime.now().strftime("%Y-%m-%d %H:00 ") + headline + fileending
elif len(args) == 0:
selection = selectfile()
inputfile = selection[0]
outputfile = selection[1] + fileending
headline = selection[2]
elif len(args) > 1:
sys.exit(red + "Too much or wrong parameters:\n" + yellow + myname + ' [--html] [--stdout] ["Some Topic"] ')
# This is informal asking yes/no for editing the text
if outfile == sys.stdout:
outputfile = "STDOUT"
print(yellow + "\nYou will edit now: '" + green + inputfile + yellow + "'\nOutputfile= '" + green + outputfile + "'")
a = input("\nPress to continue or 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 = (
'\n\
\n'
+ headline
+ '\n\
\n\
\n\
\n\
'
)
# Some fancy css for minimalistic terminal style
style_fn = ''
head2 = "\n"
body1 = 'Back
Moin'
foot = "
\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)
# Dig the files-directory and generate an index.
linecounter=mkdirindex(filesdir, indexfilename, filesdirheadline)
print("Dirindex in ",filesdir," has ", linecounter, "lines")
# Have a nice time.