141 lines
4.8 KiB
Python
141 lines
4.8 KiB
Python
import datetime
|
|
import html
|
|
import os.path
|
|
import shutil
|
|
import sys
|
|
from pathlib import Path
|
|
import markdown
|
|
from bs4 import BeautifulSoup
|
|
|
|
import find_ini
|
|
import tagging
|
|
|
|
|
|
def publish_path(article_path):
|
|
if os.path.exists(article_path):
|
|
parse_file(article_path)
|
|
else:
|
|
print("The given path of {}\nWas not found.".format(os.path.abspath(article_path)))
|
|
|
|
|
|
def publish_parse():
|
|
md_files = list(Path('.').glob('*.md'))
|
|
md_article = choose_option(md_files, "file to publish")
|
|
parse_file(md_article)
|
|
|
|
|
|
def parse_file(article_path):
|
|
blog_config = find_ini.return_options()
|
|
web_root = blog_config['general']['web_root']
|
|
sub_folder = blog_config['general']['sub_folder']
|
|
print("Parsing {}".format(os.path.abspath(article_path)))
|
|
title_from_header = find_header(os.path.abspath(article_path))
|
|
title_from_path = find_filename(os.path.abspath(article_path))
|
|
final_title = choose_option([title_from_path, title_from_header], "title")
|
|
print("The selected title is: {0}".format(final_title))
|
|
date_now = datetime.datetime.now()
|
|
year = date_now.strftime("%Y")
|
|
month = date_now.strftime("%m")
|
|
index_check_path = os.path.join(web_root, sub_folder, 'by_year')
|
|
path_to_publish = os.path.join(index_check_path, year, month)
|
|
article_number = next_article_num(path_to_publish)
|
|
path_to_publish = os.path.join(path_to_publish, article_number + final_title.replace(' ', '_'))
|
|
print("Publishing to: {0}".format(path_to_publish))
|
|
os.makedirs(path_to_publish)
|
|
index_file = os.path.join(blog_config['general']['web_root'], 'common', 'php', 'index.php')
|
|
place_indexes(index_check_path, path_to_publish, index_file)
|
|
add_datestamp(article_path, os.path.join(path_to_publish, "article.md"))
|
|
article_imgs = find_local_links(article_path)
|
|
copy_all_imgs(article_imgs, path_to_publish)
|
|
tagging.tag_article(path_to_publish)
|
|
|
|
|
|
def place_indexes(root_of_index, path_to_index, index_file):
|
|
while path_to_index != root_of_index:
|
|
print(path_to_index)
|
|
if not os.path.exists(os.path.join(path_to_index, "index.php")):
|
|
os.symlink(index_file, os.path.join(path_to_index, "index.php"))
|
|
path_to_index = os.path.dirname(path_to_index)
|
|
|
|
|
|
def find_header(article_path):
|
|
with open(article_path, "r") as article_source:
|
|
article_text = article_source.read()
|
|
article_html = markdown.markdown(article_text)
|
|
soup = BeautifulSoup(article_html, 'html.parser')
|
|
return soup.h1.string
|
|
|
|
|
|
def find_local_links(article_path):
|
|
article_imgs = []
|
|
with open(article_path, "r") as article_source:
|
|
article_text = article_source.read()
|
|
article_html = markdown.markdown(article_text)
|
|
soup = BeautifulSoup(article_html, 'html.parser')
|
|
for local_link in soup.find_all('a'):
|
|
escaped_link = html.unescape(local_link.get('href'))
|
|
if is_local(escaped_link):
|
|
print(escaped_link)
|
|
article_imgs.append(escaped_link)
|
|
for imgs in soup.find_all('img'):
|
|
print(imgs.get('src'))
|
|
article_imgs.append(imgs.get('src'))
|
|
return article_imgs
|
|
|
|
|
|
def is_local(file_href):
|
|
remote_extensions = ["http://", "https://", "ftp://", "tel:", "mailto:"]
|
|
for rx in remote_extensions:
|
|
if file_href.startswith(rx):
|
|
return False
|
|
return True
|
|
|
|
|
|
def add_datestamp(article_path, output_path):
|
|
publish_time = "Published: " + datetime.datetime.now().strftime("%Y/%m/%d")
|
|
with open(article_path, "r") as file_line:
|
|
file_contents = file_line.readlines()
|
|
file_contents.insert(2, "<p id=\"date\">" + publish_time + "</p>\n")
|
|
with open(output_path, "w") as file_lines:
|
|
file_contents = "".join(file_contents)
|
|
file_lines.write(file_contents)
|
|
|
|
|
|
def copy_all_imgs(article_imgs, path_to_publish):
|
|
for img in article_imgs:
|
|
img_src = os.path.join('.', img)
|
|
img_dest = os.path.join(path_to_publish, img)
|
|
if not os.path.exists(os.path.dirname(img_dest)):
|
|
os.makedirs(os.path.dirname(img_dest))
|
|
shutil.copyfile(img_src, img_dest)
|
|
|
|
|
|
def find_filename(article_path):
|
|
base_name = os.path.basename(article_path)
|
|
return os.path.splitext(base_name)[0]
|
|
|
|
|
|
def next_article_num(source_dir):
|
|
source_path = Path(source_dir)
|
|
number_of_dirs = 0
|
|
for file in source_path.glob('*'):
|
|
if file.is_dir():
|
|
number_of_dirs += 1
|
|
next_num = str(number_of_dirs + 1)
|
|
return next_num.zfill(2) + "_"
|
|
|
|
|
|
def choose_option(options, selecting):
|
|
count = 1
|
|
for option in options:
|
|
print("{0}. {1}".format(count, option))
|
|
count += 1
|
|
else:
|
|
print("q. exit")
|
|
option_selection = input("Select a " + selecting + ": ")
|
|
if option_selection.lower() == "q":
|
|
sys.exit()
|
|
else:
|
|
option_selection = int(option_selection)
|
|
return options[option_selection - 1]
|