def run():
parser = argparse.ArgumentParser(description='Update a website from a spreadsheet. '
'Take a spreadsheet (from google sheets or locally), and '
'convert it to a .json file that a static website '
'generator like jekyll can use, then push it out.')
parser.add_argument('layout_file', nargs="?", help='json file '
'describing source, destination, and all settings')
parser.add_argument('--cache', nargs=1, required=False, default='cache',
help='cache directory where work is stored.')
parser.add_argument('--spool', nargs=1, required=False,
help='if supplied, work only on sheets mentioned in this directory.'
'(see sheetmail)')
args = parser.parse_args()
if args.layout_file is None:
print "Need a layout file, I should give you an example."
print "See example_sites.json in github repository for sheetsite."
print "Add -h for help."
exit(1)
layout = json.loads(open(args.layout_file).read())
root = args.cache[0]
spool = args.spool[0]
names = layout['names']
for name in names:
site = layout['sites'][name]
source = site['source']
if source['name'] != 'google-sheets':
print "do not know how to read from", source['name']
exit(1)
if spool is not None:
key = source['key']
pending_file = os.path.join(spool, '{}.pending.json'.format(key))
processing_file = os.path.join(spool, '{}.processing.json'.format(key))
present = False
if os.path.exists(pending_file):
shutil.move(pending_file, processing_file)
present = True
if os.path.exists(processing_file):
present = True
if not present:
continue
path = os.path.join(root, name)
if not(os.path.exists(path)):
os.makedirs(path)
from sheetsite.google_spreadsheet import GoogleSpreadsheet
from sheetsite.site import Site
wb = GoogleSpreadsheet()
wb.connect(source['credential_file'])
wb.load_remote(source['key'])
ss = Site(wb, os.path.join(path, 'geocache.sqlite'))
if 'flags' in site:
ss.configure(site['flags'])
output_file = os.path.join(path, 'public.json')
private_output_file = os.path.join(path, 'private.json')
ss.save_local(output_file)
ss.save_local(private_output_file, private_sheets=True)
destination = site['destination']
if destination['name'] != 'git':
print "do not know how to write to", destination['name']
exit(1)
local_repo = os.path.join(path, 'repo')
if not(os.path.exists(local_repo)):
subprocess.check_output(['git', 'clone', destination['repo'], local_repo])
wd = os.getcwd()
os.chdir(local_repo)
subprocess.check_output(['git', 'pull'])
os.chdir(wd)
shutil.copyfile(output_file, os.path.join(local_repo, destination['file']))
os.chdir(local_repo)
subprocess.check_output(['git', 'add', destination['file']])
try:
subprocess.check_output(['git', 'commit', '-m', 'update from sheetsite'])
subprocess.check_output(['git', 'push'])
except subprocess.CalledProcessError:
print "Commit/push skipped"
os.chdir(wd)
if spool is not None:
key = source['key']
processing_file = os.path.join(spool, '{}.processing.json'.format(key))
ack_pending_file = os.path.join(spool, '{}.ack_pending.json'.format(key))
if os.path.exists(processing_file):
shutil.move(processing_file, ack_pending_file)
def apply_chain(site, path):
if not (os.path.exists(path)):
os.makedirs(path)
source = site['source']
destination = site['destination']
tweaks = site.get('tweaks')
wb = None
raw_file = os.path.join(path, 'raw.json')
if 'cache' in source:
wb = read_source({'filename': raw_file})
else:
wb = read_source(source)
ss = Site(wb, os.path.join(path, 'geocache.sqlite'))
if 'flags' in site:
ss.configure(site['flags'])
output_file = os.path.join(path, 'public.json')
prev_raw_file = os.path.join(path, 'prev_raw.json')
private_output_file = os.path.join(path, 'private.json')
id_file = os.path.join(path, 'ids.json')
prev_id_file = os.path.join(path, 'prev_ids.json')
if os.path.exists(raw_file):
shutil.copyfile(raw_file, prev_raw_file)
if os.path.exists(id_file):
shutil.copyfile(id_file, prev_id_file)
ss.save_local(raw_file, enhance=False)
ids = process_ids(prev_raw_file, raw_file, prev_id_file, id_file)
ss.add_ids(ids)
state = {'path': path, 'output_file': output_file, 'id_file': id_file}
if tweaks:
import json
wj = json.load(open(raw_file, 'r'))
for tweak, params in tweaks.items():
print("Working on tweak", json.dumps(tweak))
if 'tweak' in params:
tweak = params['tweak']
import importlib
mod = importlib.import_module('sheetsite.tweaks.{}'.format(tweak))
ct = 2
try:
target = mod.apply3
ct = 3
except AttributeError:
target = mod.apply
if ct == 3:
target(wj, params, state)
else:
target(wj, params)
from sheetsite.json_spreadsheet import JsonSpreadsheet
ss.workbook = JsonSpreadsheet(None, data=wj)
ss.save_local(output_file)
if not os.path.exists(prev_raw_file):
# once daff can cope with blank tables correctly, switch to this
# with open(prev_raw_file, 'w') as fout:
# fout.write('{ "names": [], "tables": [] }')
shutil.copyfile(raw_file, prev_raw_file)
shutil.copyfile(id_file, prev_id_file)
ss.save_local(private_output_file, private_sheets=True)
state['workbook'] = ss.public_workbook()
write_destination(destination, state)
return {'prev_raw_file': prev_raw_file, 'raw_file': raw_file}