diff --git a/publish_blog.py b/publish_blog.py index e69de29..846455c 100644 --- a/publish_blog.py +++ b/publish_blog.py @@ -0,0 +1,163 @@ +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', '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) + menu_path = os.path.join(web_root, "common", "menu.php") + update_menu(menu_path, year, month) + 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, "

" + publish_time + "

\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 update_menu(menu_path, year, month): + print(menu_path) + with open(menu_path, "r") as menu_source: + menu_text = menu_source.read() + soup = BeautifulSoup(menu_text, 'html.parser') + list_items = soup.find_all(id="blog") + last_post = soup.find_all(id="current_year") + # current_date = soup.find_all(attrs={"id": "date"}) + item_count = 0 + for items in last_post: + item_count += 1 + print(item_count) + print(type(items)) + print(items.attrs) + if "date=" in items: + print(items) + # print(list_items) + # print(last_post) + # print(current_date) + + +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]