/
lektor_netlify.py
79 lines (66 loc) · 2.55 KB
/
lektor_netlify.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
# -*- coding: utf-8 -*-
import click
import requests
from lektor.pluginsystem import Plugin
from lektor.publisher import Publisher, Command
from lektor.utils import slugify, bool_from_string
from werkzeug import urls
class NetlifyPublisher(Publisher):
def publish(self, target_url, credentials=None, server_info=None, **extra):
draft = '--draft' in click.get_current_context().args
host = target_url.host
if credentials and credentials.get('key'):
access_token = credentials['key']
elif server_info and server_info.extra.get('key'):
access_token = server_info.extra['key']
else:
raise RuntimeError(
"Use lektor deploy --key <ACCESS_TOKEN>,"
" see https://github.com/ajdavis/lektor-netlify/README.md")
sites_url = (
'https://api.netlify.com/api/v1/sites?access_token=' +
access_token)
response = requests.get(
sites_url,
headers={'User-Agent': 'https://github.com/ajdavis/lektor-netlify'})
response.raise_for_status()
j = response.json()
for site in j:
site_url = urls.url_parse(unicode(site['url']))
if site_url.host == host:
site_id = site['site_id']
break
else:
site_name = slugify(host).replace('.', '-')
force_ssl = bool_from_string(server_info.extra.get('force_ssl'),
default=False)
print('Creating new Netlify site "%s" at %s' % (site_name, host))
response = requests.post(sites_url, {
'name': site_name,
'custom_domain': host,
'force_ssl': force_ssl})
response.raise_for_status()
site_id = response.json()['site_id']
cmd = [
'netlify',
'deploy',
'--auth', access_token,
'--site', site_id,
'--dir', self.output_path]
if draft:
yield "Deploying as draft"
else:
cmd.append('--prod')
for line in Command(cmd):
yield line
class NetlifyPlugin(Plugin):
name = u'Netlify'
description = u'Lektor plugin to publish your site with Netlify.'
def on_setup_env(self, **extra):
try:
# Lektor 2.0+.
self.env.add_publisher('netlify', NetlifyPublisher)
except AttributeError:
# Lektor 1.0.
from lektor.publisher import publishers
publishers['netlify'] = NetlifyPublisher