def init_xsrf_secret_if_none(cls):
"""Verifies that non-default XSRF secret exists; creates one if not."""
# Any non-default value is fine.
if XSRF_SECRET.value and XSRF_SECRET.value != XSRF_SECRET.default_value:
return
# All property manipulations must run in the default namespace.
old_namespace = namespace_manager.get_namespace()
try:
namespace_manager.set_namespace(
appengine_config.DEFAULT_NAMESPACE_NAME)
# Look in the datastore directly.
entity = ConfigPropertyEntity.get_by_key_name(XSRF_SECRET.name)
if not entity:
entity = ConfigPropertyEntity(key_name=XSRF_SECRET.name)
# Any non-default non-None value is fine.
if (entity.value and not entity.is_draft and
(str(entity.value) != str(XSRF_SECRET.default_value))):
return
# Initialize to random value.
entity.value = base64.urlsafe_b64encode(
os.urandom(XSRF_SECRET_LENGTH))
entity.is_draft = False
entity.put()
finally:
namespace_manager.set_namespace(old_namespace)
def init_xsrf_secret_if_none(cls):
"""Verifies that non-default XSRF secret exists; creates one if not."""
# Any non-default value is fine.
if XSRF_SECRET.value and XSRF_SECRET.value != XSRF_SECRET.default_value:
return
# All property manipulations must run in the default namespace.
old_namespace = namespace_manager.get_namespace()
try:
namespace_manager.set_namespace(
appengine_config.DEFAULT_NAMESPACE_NAME)
# Look in the datastore directly.
entity = ConfigPropertyEntity.get_by_key_name(XSRF_SECRET.name)
if not entity:
entity = ConfigPropertyEntity(key_name=XSRF_SECRET.name)
# Any non-default non-None value is fine.
if (entity.value and not entity.is_draft and
(str(entity.value) != str(XSRF_SECRET.default_value))):
return
# Initialize to random value.
entity.value = base64.urlsafe_b64encode(
os.urandom(XSRF_SECRET_LENGTH))
entity.is_draft = False
entity.put()
finally:
namespace_manager.set_namespace(old_namespace)
def _add_new_course_entry_to_persistent_configuration(raw):
"""Adds new raw course entry definition to the datastore settings.
This loads all current datastore course entries and adds a new one. It
also find the best place to add the new entry at the further down the list
the better, because entries are applied in the order of declaration.
Args:
raw: The course entry rule: 'course:/foo::ns_foo'.
Returns:
True if added, False if not. False almost always means a duplicate rule.
"""
# Get all current entries from a datastore.
entity = ConfigPropertyEntity.get_by_key_name(GCB_COURSES_CONFIG.name)
if not entity:
entity = ConfigPropertyEntity(key_name=GCB_COURSES_CONFIG.name)
entity.is_draft = False
if not entity.value:
entity.value = GCB_COURSES_CONFIG.value
lines = entity.value.splitlines()
# Add new entry to the rest of the entries. Since entries are matched
# in the order of declaration, try to find insertion point further down.
final_lines_text = None
for index in reversed(range(0, len(lines) + 1)):
# Create new rule list putting new item at index position.
new_lines = lines[:]
new_lines.insert(index, raw)
new_lines_text = '\n'.join(new_lines)
# Validate the rule list definition.
errors = []
_courses_config_validator(new_lines_text, errors)
if not errors:
final_lines_text = new_lines_text
break
# Save updated course entries.
if final_lines_text:
entity.value = final_lines_text
entity.put()
return True
return False