pangea
/
django-orchestra
7
0
Fork 0
Code Issues 1 Pull Requests 2 Packages Projects Releases Wiki Activity

fixed

This commit is contained in:
Cayo Puigdefabregas 2021-01-30 13:55:27 +01:00
parent 7c62092faa
commit 78db4fb8d5
1 changed files with 51 additions and 51 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
orchestra/management/commands/setuppostgres.py
View File

@ -11,50 +11,52 @@ from orchestra.utils.sys import run, check_root
class Command(BaseCommand):
help = 'Setup PostgreSQL database.'
def __init__(self, *args, **kwargs):
super(Command, self).__init__(*args, **kwargs)
# Get defaults from settings, if exists
try:
defaults = settings.DATABASES['default']
self.defaults = settings.DATABASES['default']
except (AttributeError, KeyError):
defaults = {}
else:
if defaults['ENGINE'] != 'django.db.backends.postgresql_psycopg2':
defaults = {}
if self.defaults['ENGINE'] != 'django.db.backends.postgresql_psycopg2':
self.defaults = {}
def add_arguments(self, parser):
parser.add_argument(
'--db_name',
dest='db_name',
default=defaults.get('DB_NAME', 'orchestra'),
default=self.defaults.get('DB_NAME', 'orchestra'),
help='Specifies the database to create.',
type=str
)
parser.add_argument(
'--db_user',
dest='db_user',
default=defaults.get('DB_USER', 'orchestra'),
default=self.defaults.get('DB_USER', 'orchestra'),
help='Specifies the database to create.',
type=str
)
parser.add_argument(
'--db_password',
dest='db_password',
default=defaults.get('PASSWORD', ''),
default=self.defaults.get('PASSWORD', ''),
help='Specifies the database password, random if not specified.',
type=str
)
parser.add_argument(
'--db_host',
dest='db_host',
default=defaults.get('HOST', 'localhost'),
default=slef.defaults.get('HOST', 'localhost'),
help='Specifies the database to create.',
type=str
)
parser.add_argument(
'--db_port',
dest='db_port',
default=defaults.get('PORT', '5432'),
default=self.defaults.get('PORT', '5432'),
help='Specifies the database to create.',
type=str
)
@ -69,8 +71,6 @@ class Command(BaseCommand):
type=str
)
help = 'Setup PostgreSQL database.'
def run_postgres(self, cmd, *args, **kwargs):
return run('su postgres -c "psql -c \\"%s\\""' % cmd, *args, **kwargs)