[rql rewriting] handle case where we've and optional inlined relation in the original query. Also, we should append EXISTS even in subquery to avoid inserting duplicates in results
"""some utility functions for datastore initialization.
:organization: Logilab
:copyright: 2008-2010 LOGILAB S.A. (Paris, FRANCE), license is LGPL v2.
:contact: http://www.logilab.fr/ -- mailto:contact@logilab.fr
:license: GNU Lesser General Public License, v2.1 - http://www.gnu.org/licenses
"""
__docformat__ = "restructuredtext en"
from google.appengine.api.datastore import Key, Entity, Put, Get, Query
from google.appengine.api import datastore_errors
_GROUP_CACHE = {} # XXX use memcache
def _get_group(groupname):
try:
return _GROUP_CACHE[groupname]
except KeyError:
key = Key.from_path('CWGroup', 'key_' + groupname, parent=None)
try:
group = Get(key)
except datastore_errors.EntityNotFoundError:
raise Exception('can\'t find required group %s, is your instance '
'correctly initialized (eg did you run the '
'initialization script) ?' % groupname)
_GROUP_CACHE[groupname] = group
return group
def create_user(login, password, groups):
"""create a cubicweb user"""
from cubicweb.server.utils import crypt_password
user = Entity('CWUser', name=login)
user['s_login'] = unicode(login)
user['s_upassword'] = crypt_password(password)
set_user_groups(user, groups)
Put(user)
return user
def create_groups():
"""create initial cubicweb groups"""
for groupname in ('managers', 'users', 'guests'):
group = Entity('CWGroup', name='key_' + groupname)
group['s_name'] = unicode(groupname)
Put(group)
_GROUP_CACHE[groupname] = group
def set_user_groups(user, groups):
"""set user in the given groups (as string). The given user entity
(datastore.Entity) is not putted back to the repository, this is the caller
responsability.
"""
groups = [_get_group(g) for g in groups]
user['s_in_group'] = [g.key() for g in groups] or None
for group in groups:
try:
group['o_in_group'].append(user.key())
except (KeyError, AttributeError):
group['o_in_group'] = [user.key()]
Put(group)
def init_relations(gaeentity, eschema):
"""set None for every subject relations which is not yet defined"""
for rschema in eschema.subject_relations():
if rschema in ('identity', 'has_text'):
continue
dsrelation = 's_' + rschema.type
if not dsrelation in gaeentity:
gaeentity[dsrelation] = None
for rschema in eschema.object_relations():
if rschema == 'identity':
continue
dsrelation = 'o_' + rschema.type
if not dsrelation in gaeentity:
gaeentity[dsrelation] = None
def fix_entities(schema):
for etype in ('CWUser', 'CWGroup'):
eschema = schema.eschema(etype)
for gaeentity in Query(etype).Run():
init_relations(gaeentity, eschema)
# XXX o_is on CWEType entity
gaeentity['s_is'] = Key.from_path('CWEType', 'key_' + etype, parent=None)
Put(gaeentity)
def init_persistent_schema(ssession, schema):
execute = ssession.execute
rql = ('INSERT CWEType X: X name %(name)s, X description %(descr)s,'
'X final FALSE')
eschema = schema.eschema('CWEType')
execute(rql, {'name': u'CWEType', 'descr': unicode(eschema.description)})
for eschema in schema.entities():
if eschema.final or eschema == 'CWEType':
continue
execute(rql, {'name': unicode(eschema),
'descr': unicode(eschema.description)})
def insert_versions(ssession, config):
execute = ssession.execute
# insert versions
execute('INSERT CWProperty X: X pkey %(pk)s, X value%(v)s',
{'pk': u'system.version.cubicweb',
'v': unicode(config.cubicweb_version())})
for cube in config.cubes():
execute('INSERT CWProperty X: X pkey %(pk)s, X value%(v)s',
{'pk': u'system.version.%s' % cube,
'v': unicode(config.cube_version(cube))})