import os import json from datetime import datetime from string import Template from config import ( INDEX_TEMPLATE, INDEX_ROW_TEMPLATE, LINK_INDEX_TEMPLATE, ARCHIVE_PERMISSIONS, ARCHIVE_DIR, ANSI, GIT_SHA, ) from util import ( chmod_file, html_appended_url, derived_link_info, ) ### Homepage index for all the links def write_links_index(out_dir, links): """create index.html file for a given list of links""" if not os.path.exists(out_dir): os.makedirs(out_dir) print('[i] [{}] Updating {}{}{} links in archive index...'.format( datetime.now().strftime('%Y-%m-%d %H:%M:%S'), ANSI['green'], len(links), ANSI['reset'], )) write_json_links_index(out_dir, links) write_html_links_index(out_dir, links) def write_json_links_index(out_dir, links): """write the json link index to a given path""" path = os.path.join(out_dir, 'index.json') index_json = { 'info': 'Bookmark Archiver Index', 'help': 'https://github.com/pirate/bookmark-archiver', 'version': GIT_SHA, 'num_links': len(links), 'updated': str(datetime.now().timestamp()), 'links': links, } with open(path, 'w', encoding='utf-8') as f: json.dump(index_json, f, indent=4, default=str) chmod_file(path) def parse_json_links_index(out_dir): """load the index in a given directory and merge it with the given link""" index_path = os.path.join(out_dir, 'index.json') if os.path.exists(index_path): with open(index_path, 'r', encoding='utf-8') as f: return json.load(f)['links'] return [] def write_html_links_index(out_dir, links): """write the html link index to a given path""" path = os.path.join(out_dir, 'index.html') with open(INDEX_TEMPLATE, 'r', encoding='utf-8') as f: index_html = f.read() with open(INDEX_ROW_TEMPLATE, 'r', encoding='utf-8') as f: link_row_html = f.read() link_rows = '\n'.join( Template(link_row_html).substitute(**derived_link_info(link)) for link in links ) template_vars = { 'num_links': len(links), 'date_updated': datetime.now().strftime('%Y-%m-%d'), 'time_updated': datetime.now().strftime('%Y-%m-%d %H:%M'), 'rows': link_rows, } with open(path, 'w', encoding='utf-8') as f: f.write(Template(index_html).substitute(**template_vars)) chmod_file(path) ### Individual link index def write_link_index(out_dir, link): link['updated'] = str(datetime.now().timestamp()) write_json_link_index(out_dir, link) write_html_link_index(out_dir, link) def write_json_link_index(out_dir, link): """write a json file with some info about the link""" path = os.path.join(out_dir, 'index.json') print(' √ Updating: index.json') with open(path, 'w', encoding='utf-8') as f: json.dump(link, f, indent=4, default=str) chmod_file(path) def parse_json_link_index(out_dir): """load the json link index from a given directory""" existing_index = os.path.join(out_dir, 'index.json') if os.path.exists(existing_index): with open(existing_index, 'r', encoding='utf-8') as f: return json.load(f) return {} def write_html_link_index(out_dir, link): with open(LINK_INDEX_TEMPLATE, 'r', encoding='utf-8') as f: link_html = f.read() path = os.path.join(out_dir, 'index.html') print(' √ Updating: index.html') with open(path, 'w', encoding='utf-8') as f: f.write(Template(link_html).substitute({ **link, **link['latest'], 'type': link['type'] or 'website', 'tags': link['tags'] or 'untagged', 'bookmarked': datetime.fromtimestamp(float(link['timestamp'])).strftime('%Y-%m-%d %H:%M'), 'updated': datetime.fromtimestamp(float(link['updated'])).strftime('%Y-%m-%d %H:%M'), 'archive_org': link['latest'].get('archive_org') or 'https://web.archive.org/save/{}'.format(link['url']), 'wget': link['latest'].get('wget') or link['domain'], })) chmod_file(path)