def load_fixtures():
# can't pass *.json to subprocess, so these 3 lines do *.json
fixture_list = filter(lambda fn: fn.endswith('.json'),
os.listdir(os.path.join(tasklib.env['django_dir'], 'fixtures', 'initial_data')))
fixture_list = map(lambda fn: os.path.join('fixtures', 'initial_data', fn),
fixture_list)
fixture_list.sort()
tasklib._manage_py(['loaddata'] + fixture_list)
def load_admin_user(environment):
"""load admin user based on environment. """
# TODO: maybe check if the admin user already exists before replacing it ...
local_fixtures_directory = os.path.join(tasklib.env["django_dir"], "publicweb", "fixtures")
admin_user_path = os.path.join(local_fixtures_directory, environment + "_admin_user.json")
if os.path.exists(admin_user_path):
tasklib._manage_py(["loaddata", admin_user_path])
else:
tasklib._manage_py(["loaddata", "default_admin_user.json"])
def load_sample_data(environment, force=False):
"""load sample data if required."""
if force == False:
# first check if it has already been loaded
output_lines = tasklib._manage_py(['dumpdata', 'publicweb'], supress_output=True)
if output_lines[0] != '[]':
return
tasklib._manage_py(['loaddata', "sample_data.json"])
def load_sample_data(environment, force=False):
"""load sample data if required."""
if force == False:
# first check if it has already been loaded
output_lines = tasklib._manage_py(["dumpdata", "publicweb"])
if output_lines[0] != "[]":
print "Environment '", environment, "' already has sample data loaded."
return
tasklib._manage_py(["loaddata", "sample_data.json"])
def _manage_py_jenkins(apps_to_test):
""" run the jenkins command """
args = ['jenkins', ]
args += ['--pylint-rcfile', os.path.join(tasklib.env['project_dir'], 'jenkins', 'pylint.rc')]
coveragerc_filepath = os.path.join(tasklib.env['project_dir'], 'jenkins', 'coverage.rc')
if os.path.exists(coveragerc_filepath):
args += ['--coverage-rcfile', coveragerc_filepath]
args += apps_to_test
if not tasklib.env['quiet']:
print "### Running django-jenkins, with args; %s" % args
tasklib._manage_py(args, cwd=tasklib.env['project_dir'])
def load_admin_user(environment):
"""load admin user based on environment. """
# TODO: maybe check if the admin user already exists before replacing it ...
local_fixtures_directory = os.path.join(tasklib.env['django_dir'],
'publicweb', 'fixtures')
admin_user_path = os.path.join(local_fixtures_directory,
environment + '_admin_user.json')
if os.path.exists(admin_user_path):
tasklib._manage_py(['loaddata', admin_user_path])
else:
tasklib._manage_py(['loaddata', "default_admin_user.json"])
def load_fixtures():
# can't pass *.json to subprocess, so these 3 lines do *.json
fixture_list = filter(
lambda fn: fn.endswith('.json'),
os.listdir(
os.path.join(tasklib.env['django_dir'], 'fixtures',
'initial_data')))
fixture_list = map(lambda fn: os.path.join('fixtures', 'initial_data', fn),
fixture_list)
fixture_list.sort()
_move_to_end(fixture_list,
"fixtures/initial_data/product_registrations.json")
tasklib._manage_py(['loaddata'] + fixture_list)
def load_django_site_data(environment, force=False):
"""load data for django sites framework. """
if force == False:
# first check if it has already been loaded
output_lines = tasklib._manage_py(["dumpdata", "sites"])
if output_lines[0] != "[]":
print "Environment '", environment, "' already has site data loaded."
return
local_fixtures_directory = os.path.join(tasklib.env["django_dir"], "publicweb", "fixtures")
site_fixture_path = os.path.join(local_fixtures_directory, environment + "_site.json")
if os.path.exists(site_fixture_path):
tasklib._manage_py(["loaddata", site_fixture_path])
else:
tasklib._manage_py(["loaddata", "default_site.json"])
def load_django_site_data(environment, force=False):
"""load data for django sites framework. """
if force == False:
# first check if it has already been loaded
output_lines = tasklib._manage_py(['dumpdata', 'sites'], supress_output=True)
if output_lines[0] != '[]':
return
local_fixtures_directory = os.path.join(tasklib.env['django_dir'], 'publicweb',
'fixtures')
site_fixture_path = os.path.join(local_fixtures_directory,
environment +'_site.json')
if os.path.exists(site_fixture_path):
tasklib._manage_py(['loaddata', site_fixture_path])
else:
tasklib._manage_py(['loaddata', "default_site.json"])
def load_django_site_data(environment, force=False):
"""Load data for django sites framework. """
if force == False:
site_needs_initializing = int(tasklib._manage_py(['site_needs_initializing'])[0].strip())
if not site_needs_initializing:
print "Environment '", environment, "' already has site data initialized."
return
local_fixtures_directory = os.path.join(tasklib.env['django_dir'], 'publicweb',
'fixtures')
site_fixture_path = os.path.join(local_fixtures_directory,
environment +'_site.json')
if os.path.exists(site_fixture_path):
tasklib._manage_py(['loaddata', site_fixture_path])
else:
tasklib._manage_py(['loaddata', "default_site.json"])
def load_django_site_data(environment, force=False):
"""load data for django sites framework. """
if force == False:
# first check if it has already been loaded
output_lines = tasklib._manage_py(['dumpdata', 'sites'], supress_output=True)
if output_lines[0] != '[]':
return
local_fixtures_directory = os.path.join(tasklib.env['django_dir'], 'publicweb',
'fixtures')
site_fixture_path = os.path.join(local_fixtures_directory,
environment +'_site.json')
if os.path.exists(site_fixture_path):
tasklib._manage_py(['loaddata', site_fixture_path])
else:
tasklib._manage_py(['loaddata', "default_site.json"])
def load_auth_user(environment, force=False):
"""load auth user fixture based on environment. """
if force == False:
auth_user_needs_initializing = int(tasklib._manage_py(['auth_user_needs_initializing'])[0].strip())
if not auth_user_needs_initializing:
print "Environment '", environment, "' already has auth.user initialized."
return
local_fixtures_directory = os.path.join(tasklib.env['django_dir'], 'publicweb',
'fixtures')
user_path = os.path.join(local_fixtures_directory,
environment +'_auth_user.json')
if os.path.exists(user_path):
tasklib._manage_py(['loaddata', user_path])
else:
tasklib._manage_py(['loaddata', "default_auth_user.json"])
def load_django_site_data(environment, force=False):
"""Load data for django sites framework. """
if force == False:
site_needs_initializing = int(
tasklib._manage_py(['site_needs_initializing'])[0].strip())
if not site_needs_initializing:
print "Environment '", environment, "' already has site data initialized."
return
local_fixtures_directory = os.path.join(tasklib.env['django_dir'],
'publicweb', 'fixtures')
site_fixture_path = os.path.join(local_fixtures_directory,
environment + '_site.json')
if os.path.exists(site_fixture_path):
tasklib._manage_py(['loaddata', site_fixture_path])
else:
tasklib._manage_py(['loaddata', "default_site.json"])
def load_auth_user(environment, force=False):
"""load auth user fixture based on environment. """
if force == False:
auth_user_needs_initializing = int(
tasklib._manage_py(['auth_user_needs_initializing'])[0].strip())
if not auth_user_needs_initializing:
print "Environment '", environment, "' already has auth.user initialized."
return
local_fixtures_directory = os.path.join(tasklib.env['django_dir'],
'publicweb', 'fixtures')
user_path = os.path.join(local_fixtures_directory,
environment + '_auth_user.json')
if os.path.exists(user_path):
tasklib._manage_py(['loaddata', user_path])
else:
tasklib._manage_py(['loaddata', "default_auth_user.json"])
def load_auth_user(environment, force=False):
"""load auth user fixture based on environment. """
if force == False:
# first check if it has already been loaded
output_lines = tasklib._manage_py(['dumpdata', 'auth.user'])
if output_lines[0] != '[]':
print "Environment '", environment, "' already has auth.user loaded."
return
local_fixtures_directory = os.path.join(tasklib.env['django_dir'], 'publicweb',
'fixtures')
user_path = os.path.join(local_fixtures_directory,
environment +'_auth_user.json')
if os.path.exists(user_path):
tasklib._manage_py(['loaddata', user_path])
else:
tasklib._manage_py(['loaddata', "default_auth_user.json"])
def create_cache_table():
(db_engine, db_name, db_user, db_pw, db_port) = tasklib._get_django_db_settings()
cache_table_name = 'sarpam_cache_table'
tasklib._mysql_exec('DROP TABLE IF EXISTS %s.%s' % (db_name, cache_table_name))
tasklib._manage_py(['createcachetable', cache_table_name])
from tasklib import env, _manage_py
def post_deploy(environment=None):
sys.path.append(env['django_dir'])
sys.path.append(os.path.join(env['ve_dir'], 'lib', 'python2.6', 'site-packages'))
import settings as django_settings
try:
from settings import MEDIA_ROOT
if MEDIA_ROOT is not None and MEDIA_ROOT != '':
if not os.path.exists(MEDIA_ROOT):
os.mkdir(MEDIA_ROOT)
os.chmod(MEDIA_ROOT, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
except ImportError, e:
# no MEDIA_ROOT? not a fatal error
print "No MEDIA_ROOT found: %s" % e
try:
from settings import HAYSTACK_CONNECTIONS
whoosh_path = HAYSTACK_CONNECTIONS['default']['PATH']
if not os.path.exists(whoosh_path):
os.mkdir(whoosh_path)
os.chmod(whoosh_path, stat.S_IRWXU | stat.S_IRWXG | stat.S_IRWXO)
except ImportError, e:
# no HAYSTACK_CONNECTIONS? not a fatal error
print "No HAYSTACK_CONNECTIONS found: %s" % e
_manage_py(["collectstatic", "--noinput"])
def load_sample_data():
"""load sample data (fixture) to play with"""
tasklib._manage_py(['loaddata', "sample_data.json"])
