Use upstream logic for install command, and some web menus

This commit is contained in:
Lance Edgar 2023-01-17 19:30:35 -06:00
parent 617515a167
commit c66d0ababa
6 changed files with 74 additions and 453 deletions

View file

@ -25,15 +25,10 @@ Messkit commands
"""
import os
import stat
import sys
import subprocess
import sqlalchemy as sa
from alembic.util.messaging import obfuscate_url_pw
from rattail import commands
from rattail.files import resource_path
from messkit import __version__
@ -57,203 +52,67 @@ class Command(commands.Command):
long_description = ''
class Install(commands.Subcommand):
class Install(commands.InstallSubcommand):
"""
Install a Messkit app
Install the Messkit app
"""
name = 'install'
description = __doc__.strip()
def run(self, args):
# nb. these must be explicitly set b/c config is not available
# when running normally, e.g. `messkit -n install`
app_title = "Messkit"
app_package = 'messkit'
app_eggname = 'Messkit'
app_pypiname = 'Messkit'
self.rprint("\n\t[blue]Welcome to Messkit![/blue]")
self.rprint("\n\tThis tool will install and configure a new app.")
self.rprint("\n\t[italic]NB. You should already have created a new database in PostgreSQL or MySQL.[/italic]")
def do_install_steps(self):
# continue?
if not self.basic_prompt("continue?", True, is_bool=True):
self.rprint()
sys.exit(0)
# first all normal steps
super(Install, self).do_install_steps()
# appdir must not yet exist
appdir = os.path.join(sys.prefix, 'app')
if os.path.exists(appdir):
self.rprint("\n\t[bold red]appdir already exists:[/bold red] {}\n".format(appdir))
sys.exit(1)
# we also install poser..for now..?
self.install_poser()
# get db info
dbtype = self.basic_prompt('db type', 'postgresql')
dbhost = self.basic_prompt('db host', 'localhost')
dbport = self.basic_prompt('db port', '3306' if dbtype == 'mysql' else '5432')
dbname = self.basic_prompt('db name', 'messkit')
dbuser = self.basic_prompt('db user', 'rattail')
def put_settings(self, **kwargs):
# get db password
dbpass = None
while not dbpass:
dbpass = self.basic_prompt('db pass', is_password=True)
rattail = [os.path.join(sys.prefix, 'bin', 'rattail'),
'-c', os.path.join(sys.prefix, 'app', 'silent.conf')]
# test db connection
self.rprint("\n\ttesting db connection... ", end='')
dburl = self.make_db_url(dbtype, dbhost, dbport, dbname, dbuser, dbpass)
error = self.test_db_connection(dburl)
if error:
self.rprint("[bold red]cannot connect![/bold red] ..error was:")
self.rprint("\n{}".format(error))
self.rprint("\n\t[bold yellow]aborting mission[/bold yellow]\n")
sys.exit(1)
self.rprint("[bold green]good[/bold green]")
# set falafel theme
cmd = rattail + ['setting-put', 'tailbone.theme', 'falafel']
subprocess.check_call(cmd)
# make the appdir
self.app.make_appdir(appdir)
# hide theme picker
cmd = rattail + ['setting-put', 'tailbone.themes.expose_picker', 'false']
subprocess.check_call(cmd)
# shared context for generated app files
context = {
'envdir': sys.prefix,
'app_package': 'messkit',
'app_title': "Messkit",
'appdir': appdir,
'db_url': dburl,
'pyramid_egg': 'Messkit',
'beaker_key': 'messkit',
}
# set main image
cmd = rattail + ['setting-put', 'tailbone.main_image_url', '/messkit/img/messkit.png']
subprocess.check_call(cmd)
# make config files
rattail_conf = self.app.make_config_file(
'rattail', os.path.join(appdir, 'rattail.conf'),
template_path=resource_path('messkit:templates/installer/rattail.conf.mako'),
**context)
quiet_conf = self.app.make_config_file('quiet', appdir)
web_conf = self.app.make_config_file(
'web-complete', os.path.join(appdir, 'web.conf'),
**context)
# set header image
cmd = rattail + ['setting-put', 'tailbone.header_image_url', '/messkit/img/messkit-small.png']
subprocess.check_call(cmd)
# make upgrade script
path = os.path.join(appdir, 'upgrade.sh')
self.app.render_mako_template(
resource_path('messkit:templates/installer/upgrade.sh.mako'),
context, output_path=path)
os.chmod(path, stat.S_IRWXU
| stat.S_IRGRP
| stat.S_IXGRP
| stat.S_IROTH
| stat.S_IXOTH)
# set favicon image
cmd = rattail + ['setting-put', 'tailbone.favicon_url', '/messkit/img/messkit-small.png']
subprocess.check_call(cmd)
self.rprint("\n\tappdir created at: [bold green]{}[/bold green]".format(appdir))
# set default grid page size
cmd = rattail + ['setting-put', 'tailbone.grid.default_pagesize', '20']
subprocess.check_call(cmd)
bindir = os.path.join(sys.prefix, 'bin')
schema_installed = False
if self.basic_prompt("install db schema?", True, is_bool=True):
self.rprint()
# install db schema
alembic = os.path.join(bindir, 'alembic')
cmd = [alembic, '-c', rattail_conf, 'upgrade', 'heads']
subprocess.check_call(cmd)
schema_installed = True
rattail = os.path.join(bindir, 'rattail')
# set falafel theme
cmd = [rattail, '-c', quiet_conf, '--no-versioning',
'setting-put', 'tailbone.theme', 'falafel']
subprocess.check_call(cmd)
# set main image
cmd = [rattail, '-c', quiet_conf, '--no-versioning',
'setting-put', 'tailbone.main_image_url', '/messkit/img/messkit.png']
subprocess.check_call(cmd)
# set header image
cmd = [rattail, '-c', quiet_conf, '--no-versioning',
'setting-put', 'tailbone.header_image_url', '/messkit/img/messkit-small.png']
subprocess.check_call(cmd)
# set favicon image
cmd = [rattail, '-c', quiet_conf, '--no-versioning',
'setting-put', 'tailbone.favicon_url', '/messkit/img/messkit-small.png']
subprocess.check_call(cmd)
self.rprint("\n\tdb schema installed to: [bold green]{}[/bold green]".format(
obfuscate_url_pw(dburl)))
if self.basic_prompt("create admin user?", True, is_bool=True):
# get admin credentials
username = self.basic_prompt('admin username', 'admin')
password = None
while not password:
password = self.basic_prompt('admin password', is_password=True)
if password:
confirm = self.basic_prompt('confirm password', is_password=True)
if not confirm or confirm != password:
self.rprint("[bold yellow]passwords did not match[/bold yellow]")
password = None
fullname = self.basic_prompt('full name')
self.rprint()
# make admin user
rattail = os.path.join(bindir, 'rattail')
cmd = [rattail, '-c', quiet_conf, 'make-user', '-A', username,
'--password', password]
if fullname:
cmd.extend(['--full-name', fullname])
subprocess.check_call(cmd)
self.rprint("\n\tadmin user created: [bold green]{}[/bold green]".format(
username))
if self.basic_prompt("make poser dir?", True, is_bool=True):
self.rprint()
# make poser dir
poser_handler = self.app.get_poser_handler()
poserdir = poser_handler.make_poser_dir()
self.rprint("\n\tposer dir created: [bold green]{}[/bold green]".format(
poserdir))
self.rprint("\n\t[bold green]initial setup is complete![/bold green]")
if schema_installed:
self.rprint("\n\tyou can run the web app with this command:")
pserve = os.path.join(bindir, 'pserve')
self.rprint("\n\t[blue]{} file+ini:{}[/blue]".format(pserve, web_conf))
def install_poser(self):
if not self.basic_prompt("make poser dir?", True, is_bool=True):
return False
self.rprint()
# TODO: somewhere should ask about apache proxy, https etc.?
# make poser dir
poser_handler = self.app.get_poser_handler()
poserdir = poser_handler.make_poser_dir()
def make_db_url(self, dbtype, dbhost, dbport, dbname, dbuser, dbpass):
try:
# newer style
from sqlalchemy.engine import URL
factory = URL.create
except ImportError:
# older style
from sqlalchemy.engine.url import URL
factory = URL
if dbtype == 'mysql':
drivername = 'mysql+mysqlconnector'
else:
drivername = 'postgresql+psycopg2'
return factory(drivername=drivername,
username=dbuser,
password=dbpass,
host=dbhost,
port=dbport,
database=dbname)
def test_db_connection(self, url):
engine = sa.create_engine(url)
# check for random table; does not matter if it exists, we
# just need to test interaction and this is a neutral way
try:
engine.has_table('whatever')
except Exception as error:
return str(error)
self.rprint("\n\tposer dir created: [bold green]{}[/bold green]".format(
poserdir))
return True