diff options
| author | Dirk Engling <erdgeist@erdgeist.org> | 2020-05-27 01:14:11 +0200 |
|---|---|---|
| committer | Dirk Engling <erdgeist@erdgeist.org> | 2020-05-27 01:14:11 +0200 |
| commit | 959576ad96a5a7a53cb9b9f6c5e5fbb6ecff1bda (patch) | |
| tree | d3feefda46d2f3a61a0b3eefd194e2cea72ab5bd /bin/export.py | |
| parent | daf20dd652f9294273c1110f60ff4da6d99f67c5 (diff) | |
Remove circular dependency on makefile
Diffstat (limited to 'bin/export.py')
| -rwxr-xr-x | bin/export.py | 158 |
1 files changed, 158 insertions, 0 deletions
diff --git a/bin/export.py b/bin/export.py new file mode 100755 index 00000000..26460e8f --- /dev/null +++ b/bin/export.py | |||
| @@ -0,0 +1,158 @@ | |||
| 1 | #! venv/bin/python | ||
| 2 | import psycopg2 | ||
| 3 | import psycopg2.extras | ||
| 4 | import pathlib | ||
| 5 | import os.path | ||
| 6 | import datetime | ||
| 7 | import pypandoc | ||
| 8 | import json | ||
| 9 | from datetime import timedelta, datetime, tzinfo | ||
| 10 | #from langdetect import detect | ||
| 11 | import sys | ||
| 12 | |||
| 13 | conn = psycopg2.connect(database="cccms_dev", user="postgres", password="", host="127.0.0.1") | ||
| 14 | cursor = conn.cursor(cursor_factory=psycopg2.extras.DictCursor) | ||
| 15 | lcursor = conn.cursor() | ||
| 16 | cursor.execute("select id, login, email, admin from users", []) | ||
| 17 | user = cursor.fetchall() | ||
| 18 | |||
| 19 | page_list = [] | ||
| 20 | redirect_list = {} | ||
| 21 | |||
| 22 | pathlib.Path(os.path.dirname('git/')).mkdir(parents=True, exist_ok=True) | ||
| 23 | os.system("git -C git/ init") | ||
| 24 | pathlib.Path(os.path.dirname('git/images/')).mkdir(parents=True, exist_ok=True) | ||
| 25 | pathlib.Path(os.path.dirname('git/files/')).mkdir(parents=True, exist_ok=True) | ||
| 26 | pathlib.Path(os.path.dirname('dump/images/')).mkdir(parents=True, exist_ok=True) | ||
| 27 | pathlib.Path(os.path.dirname('dump/files/')).mkdir(parents=True, exist_ok=True) | ||
| 28 | |||
| 29 | rev = 0 | ||
| 30 | cursor.execute("select id, name, upload_content_type, upload_file_name, upload_updated_at from assets order by id", []) | ||
| 31 | assets = cursor.fetchall() | ||
| 32 | for asset in assets: | ||
| 33 | if not asset.get('upload_file_name'): | ||
| 34 | continue | ||
| 35 | source = "/usr/local/www/cccms/public/system/uploads/{}/original/{}".format(asset['id'], asset['upload_file_name']) | ||
| 36 | destdir = 'files/' | ||
| 37 | if asset['upload_content_type'].startswith('image'): | ||
| 38 | destdir = 'images/' | ||
| 39 | page_list.append({ 'date': asset['upload_updated_at'], 'revision': rev, 'comment': "asset {}".format(asset['name']), 'fname': source, 'gname': destdir+asset['upload_file_name'], 'editor': 'admin', 'email': 'admin@cccms.de'}) | ||
| 40 | rev = rev + 1 | ||
| 41 | redirect_list['/public/system/uploads/{}/original/{}'.format(asset['id'], asset['upload_file_name'])] = destdir+asset['upload_file_name'] | ||
| 42 | |||
| 43 | with open('redirects.json', 'w') as outfile: | ||
| 44 | json.dump(redirect_list, outfile) | ||
| 45 | |||
| 46 | cursor.execute("select id, unique_name, created_at from nodes order by id", []) | ||
| 47 | nodes = cursor.fetchall() | ||
| 48 | |||
| 49 | for node in nodes: | ||
| 50 | if not 'unique_name' in node: | ||
| 51 | print ("WARNING: NO unique_name in node " + str(node.id)) | ||
| 52 | continue | ||
| 53 | |||
| 54 | if node['id'] == 1: | ||
| 55 | continue | ||
| 56 | |||
| 57 | if "/" in str(node['unique_name']): | ||
| 58 | pathlib.Path(os.path.dirname('dump/'+node['unique_name'])).mkdir(parents=True, exist_ok=True) | ||
| 59 | if str(node['unique_name']).startswith('updates/'): | ||
| 60 | pathlib.Path(os.path.dirname('git/'+node['unique_name'])).mkdir(parents=True, exist_ok=True) | ||
| 61 | else: | ||
| 62 | pathlib.Path(os.path.dirname('git/pages/'+node['unique_name'])).mkdir(parents=True, exist_ok=True) | ||
| 63 | |||
| 64 | cursor.execute("select id, revision, created_at, updated_at, published_at, user_id, editor_id from pages where node_id = %s", [node['id']]) | ||
| 65 | pages = cursor.fetchall() | ||
| 66 | |||
| 67 | # print ( "WORKING ON NODE: " + str(node['id'])) | ||
| 68 | |||
| 69 | for page in pages: | ||
| 70 | |||
| 71 | # ignoring unpublished pages | ||
| 72 | if not page['published_at']: | ||
| 73 | # print("UNPUBLISHED: {}\n".format(node['id'])) | ||
| 74 | continue | ||
| 75 | |||
| 76 | cursor.execute("select name from tags where id in (select tag_id from taggings where taggable_id = %s)", [page['id']]) | ||
| 77 | tags = cursor.fetchall() | ||
| 78 | |||
| 79 | editor = "admin" | ||
| 80 | if page['editor_id']: | ||
| 81 | cursor.execute("select login from users where id = %s", [page['editor_id']]) | ||
| 82 | e = cursor.fetchall() | ||
| 83 | if len(e): | ||
| 84 | editor = e[0].get("login", "NO-LOGIN") | ||
| 85 | |||
| 86 | creator = "admin" | ||
| 87 | if page['user_id']: | ||
| 88 | cursor.execute("select login from users where id = %s", [page['user_id']]) | ||
| 89 | c = cursor.fetchall() | ||
| 90 | if len(c): | ||
| 91 | creator = c[0].get("login", "NO-LOGIN") | ||
| 92 | |||
| 93 | previewimage = '' | ||
| 94 | cursor.execute("select upload_file_name from related_assets full join assets on related_assets.asset_id = assets.id where related_assets.page_id = %s and position = 1;", [page['id']]) | ||
| 95 | related_asset = cursor.fetchall() | ||
| 96 | if len(related_asset): | ||
| 97 | previewimage = related_asset[0].get('upload_file_name') | ||
| 98 | |||
| 99 | cursor.execute("select locale, title, abstract, body, created_at, updated_at from page_translations where page_id = %s", [page['id']]) | ||
| 100 | page_translations = cursor.fetchall() | ||
| 101 | for translation in page_translations: | ||
| 102 | if not translation.get("title"): | ||
| 103 | continue | ||
| 104 | |||
| 105 | if len(tags) and not isinstance(tags[0], str): | ||
| 106 | tags = [y for x in tags for y in x] | ||
| 107 | |||
| 108 | if translation['locale'] == 'en': | ||
| 109 | fname = "dump/" + node['unique_name'] + '.en.md:' + str(page['revision']) + '@' + str(int(page['updated_at'].timestamp())) | ||
| 110 | gitname = node['unique_name'] + '.en.md' | ||
| 111 | else: | ||
| 112 | fname = "dump/" + node['unique_name'] + '.md:' + str(page['revision']) + '@' + str(int(page['updated_at'].timestamp())) | ||
| 113 | gitname = node['unique_name'] + '.md' | ||
| 114 | |||
| 115 | if not gitname.startswith('updates/'): | ||
| 116 | gitname = 'pages/' + gitname | ||
| 117 | |||
| 118 | #lang = detect(translation.get('body')) | ||
| 119 | #print('{}:{} node_id {: 5d} revision {: 3d} with page_id {: 5d}, path {}'.format(lang, translation['locale'], node['id'], page['revision'], page['id'], node['unique_name'])) | ||
| 120 | |||
| 121 | # md = pypandoc.convert_text(str(translation.get('body', "")), 'md', format='html', extra_args=['--atx-headers']) | ||
| 122 | # rst = pypandoc.convert_text(str(translation.get('body', "")), 'rst', format='html', extra_args=['--atx-headers']) | ||
| 123 | # with open( "dump/" + node['unique_name'] + '.' + translation['locale'] + '.md.html:' + str(page['revision']) + '@' + str(int(page['updated_at'].timestamp())), "w") as f: | ||
| 124 | # f.write(pypandoc.convert_text( md, 'html', format='md', extra_args=['--atx-headers'])) | ||
| 125 | # with open( "dump/" + node['unique_name'] + '.' + translation['locale'] + '.rst.html:' + str(page['revision']) + '@' + str(int(page['updated_at'].timestamp())), "w") as f: | ||
| 126 | # f.write(pypandoc.convert_text( rst, 'html', format='rst', extra_args=['--atx-headers'])) | ||
| 127 | |||
| 128 | with open(fname, "w") as f: | ||
| 129 | f.write("title: {}\n".format(' '.join(translation.get("title", "").split()))) | ||
| 130 | f.write("date: {}\n".format(page['published_at'].strftime("%Y-%m-%d %H:%M:%S %z"))) | ||
| 131 | if page.get('updated_at'): | ||
| 132 | f.write("updated: {}\n".format(page['updated_at'].strftime("%Y-%m-%d %H:%M:%S %z"))) | ||
| 133 | f.write("author: {}\n".format(creator)) | ||
| 134 | f.write("tags: {}\n".format(', '.join(tags).lower())) | ||
| 135 | if previewimage: | ||
| 136 | f.write("previewimage: /images/{}\n".format(previewimage)) | ||
| 137 | f.write("\n") | ||
| 138 | |||
| 139 | # Add abstract, if one is there | ||
| 140 | if translation.get('abstract'): | ||
| 141 | f.write(str(translation.get('abstract', ""))) | ||
| 142 | f.write("\n\n") | ||
| 143 | if translation.get('body'): | ||
| 144 | f.write("<!-- TEASER_END -->\n\n") | ||
| 145 | |||
| 146 | f.write(pypandoc.convert_text( str(translation.get('body', "")), 'markdown-smart', format='html-native_divs-native_spans', extra_args=['--atx-headers'])) | ||
| 147 | userrec = next(filter(lambda person: person['login'] == editor, user)) | ||
| 148 | page_list.append({ 'date': page['updated_at'], 'revision': page['revision'], 'comment': "page revision {}".format(page['revision']), 'fname': fname, 'gname': gitname, 'editor': editor, 'email': userrec['email']}) | ||
| 149 | |||
| 150 | page_list.sort(key=lambda tup: (tup['date'], tup['revision'])) | ||
| 151 | for page in page_list: | ||
| 152 | print(page) | ||
| 153 | os.system("cp {} git/{}".format(page['fname'], page['gname'])) | ||
| 154 | os.system("git -C git/ add {}".format(page['gname'])) | ||
| 155 | os.environ['GIT_COMMITTER_NAME'] = page['editor'] | ||
| 156 | os.environ['GIT_COMMITTER_EMAIL'] = page['email'] | ||
| 157 | os.system('git -C git/ commit -m "committing {}" --author="{} <{}>" --date="{}"'.format(page['comment'], page['editor'], page['email'], page['date'])) | ||
| 158 | |||
