bitbake: toaster: Remove DATABASE_URL being passed around as an environment var

We don't need to pass the DATABASE_URL around and read it back if we
setup the django framework in the correct way.
We make the default sqlite database path a full path so that the
database isn't being assumed to be in CWD.

Also add some more useful comments on the database settings.

This is preparation work to migrate the build tests and be able to
trigger builds on differently configured databases.

(Bitbake rev: 973c740404ca6a09feea250d3433075995067fe0)

Signed-off-by: Michael Wood <michael.g.wood@intel.com>
Signed-off-by: Elliot Smith <elliot.smith@intel.com>
Signed-off-by: Richard Purdie <richard.purdie@linuxfoundation.org>
This commit is contained in:
Michael Wood 2016-05-19 13:59:28 +01:00 committed by Richard Purdie
parent 8fba59ce45
commit db84307055
4 changed files with 20 additions and 78 deletions

View File

@ -100,7 +100,6 @@ stop_system()
fi fi
webserverKillAll webserverKillAll
# unset exported variables # unset exported variables
unset DATABASE_URL
unset TOASTER_CONF unset TOASTER_CONF
unset TOASTER_DIR unset TOASTER_DIR
unset BITBAKE_UI unset BITBAKE_UI
@ -278,7 +277,6 @@ case $CMD in
return 4 return 4
fi fi
export BITBAKE_UI='toasterui' export BITBAKE_UI='toasterui'
export DATABASE_URL=`$MANAGE get-dburl`
$MANAGE runbuilds & echo $! >${BUILDDIR}/.runbuilds.pid $MANAGE runbuilds & echo $! >${BUILDDIR}/.runbuilds.pid
# set fail safe stop system on terminal exit # set fail safe stop system on terminal exit
trap stop_system SIGHUP trap stop_system SIGHUP

View File

@ -21,19 +21,19 @@ import bb
import re import re
import os import os
os.environ["DJANGO_SETTINGS_MODULE"] = "toaster.toastermain.settings"
import django import django
from django.utils import timezone from django.utils import timezone
import toaster
# Add toaster module to the search path to help django.setup() find the right
# modules
sys.path.insert(0, os.path.dirname(toaster.__file__))
def _configure_toaster(): #Set the DJANGO_SETTINGS_MODULE if it's not already set
""" Add toaster to sys path for importing modules os.environ["DJANGO_SETTINGS_MODULE"] =\
""" os.environ.get("DJANGO_SETTINGS_MODULE",
sys.path.append(os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(__file__))), 'toaster')) "toaster.toastermain.settings")
_configure_toaster() # Setup django framework (needs to be done before importing modules)
django.setup() django.setup()
from orm.models import Build, Task, Recipe, Layer_Version, Layer, Target, LogMessage, HelpText from orm.models import Build, Task, Recipe, Layer_Version, Layer, Target, LogMessage, HelpText
@ -54,11 +54,11 @@ from datetime import datetime, timedelta
from django.db import transaction, connection from django.db import transaction, connection
# pylint: disable=invalid-name # pylint: disable=invalid-name
# the logger name is standard throughout BitBake # the logger name is standard throughout BitBake
logger = logging.getLogger("ToasterLogger") logger = logging.getLogger("ToasterLogger")
class NotExisting(Exception): class NotExisting(Exception):
pass pass

View File

@ -1,9 +0,0 @@
from toastermain.settings import getDATABASE_URL
from django.core.management.base import NoArgsCommand
class Command(NoArgsCommand):
args = ""
help = "get database url"
def handle_noargs(self,**options):
print(getDATABASE_URL())

View File

@ -21,7 +21,7 @@
# Django settings for Toaster project. # Django settings for Toaster project.
import os, re import os
DEBUG = True DEBUG = True
TEMPLATE_DEBUG = DEBUG TEMPLATE_DEBUG = DEBUG
@ -38,14 +38,19 @@ ADMINS = (
MANAGERS = ADMINS MANAGERS = ADMINS
TOASTER_SQLITE_DEFAULT_DIR = os.path.join(os.environ.get('TOASTER_DIR', ''),
'build')
DATABASES = { DATABASES = {
'default': { 'default': {
'ENGINE': 'django.db.backends.sqlite3', # Add 'postgresql_psycopg2', 'mysql', 'sqlite3' or 'oracle'. # Add 'postgresql_psycopg2', 'mysql', 'sqlite3' or 'oracle'.
'NAME': 'toaster.sqlite', # Or path to database file if using sqlite3. 'ENGINE': 'django.db.backends.sqlite3',
# DB name or full path to database file if using sqlite3.
'NAME': "%s/toaster.sqlite" % TOASTER_SQLITE_DEFAULT_DIR,
'USER': '', 'USER': '',
'PASSWORD': '', 'PASSWORD': '',
'HOST': '127.0.0.1', # Empty for localhost through domain sockets or '127.0.0.1' for localhost through TCP. #'HOST': '127.0.0.1', # e.g. mysql server
'PORT': '3306', # Set to empty string for default. #'PORT': '3306', # e.g. mysql port
} }
} }
@ -55,58 +60,6 @@ DATABASES = {
if 'sqlite' in DATABASES['default']['ENGINE']: if 'sqlite' in DATABASES['default']['ENGINE']:
DATABASES['default']['OPTIONS'] = { 'timeout': 20 } DATABASES['default']['OPTIONS'] = { 'timeout': 20 }
# Reinterpret database settings if we have DATABASE_URL environment variable defined
if 'DATABASE_URL' in os.environ:
dburl = os.environ['DATABASE_URL']
if dburl.startswith('sqlite3://'):
result = re.match('sqlite3://(.*)', dburl)
if result is None:
raise Exception("ERROR: Could not read sqlite database url: %s" % dburl)
DATABASES['default'] = {
'ENGINE': 'django.db.backends.sqlite3',
'NAME': result.group(1),
'USER': '',
'PASSWORD': '',
'HOST': '',
'PORT': '',
}
elif dburl.startswith('mysql://'):
# URL must be in this form: mysql://user:pass@host:port/name
result = re.match(r"mysql://([^:]*):([^@]*)@([^:]*):(\d*)/([^/]*)", dburl)
if result is None:
raise Exception("ERROR: Could not read mysql database url: %s" % dburl)
DATABASES['default'] = {
'ENGINE': 'django.db.backends.mysql',
'NAME': result.group(5),
'USER': result.group(1),
'PASSWORD': result.group(2),
'HOST': result.group(3),
'PORT': result.group(4),
}
else:
raise Exception("FIXME: Please implement missing database url schema for url: %s" % dburl)
# Allows current database settings to be exported as a DATABASE_URL environment variable value
def getDATABASE_URL():
d = DATABASES['default']
if d['ENGINE'] == 'django.db.backends.sqlite3':
if d['NAME'] == ':memory:':
return 'sqlite3://:memory:'
elif d['NAME'].startswith("/"):
return 'sqlite3://' + d['NAME']
return "sqlite3://" + os.path.join(os.getcwd(), d['NAME'])
elif d['ENGINE'] == 'django.db.backends.mysql':
return "mysql://" + d['USER'] + ":" + d['PASSWORD'] + "@" + d['HOST'] + ":" + d['PORT'] + "/" + d['NAME']
raise Exception("FIXME: Please implement missing database url schema for engine: %s" % d['ENGINE'])
# Hosts/domain names that are valid for this site; required if DEBUG is False # Hosts/domain names that are valid for this site; required if DEBUG is False
# See https://docs.djangoproject.com/en/1.5/ref/settings/#allowed-hosts # See https://docs.djangoproject.com/en/1.5/ref/settings/#allowed-hosts
ALLOWED_HOSTS = [] ALLOWED_HOSTS = []