def test_wash_dashed_language(self):
"""messages - washing dashed language code (fr-ca)"""
if 'fr' not in cfg['CFG_SITE_LANGS']:
self.assertEqual(messages.wash_language('fr-ca'),
cfg['CFG_SITE_LANG'])
else:
self.assertEqual(messages.wash_language('fr-ca'),
'fr')
def test_wash_dashed_language(self):
"""messages - washing dashed language code (fr-ca)"""
if 'fr' not in cfg['CFG_SITE_LANGS']:
self.assertEqual(messages.wash_language('fr-ca'),
cfg['CFG_SITE_LANG'])
else:
self.assertEqual(messages.wash_language('fr-ca'),
'fr')
def format_records(records, of='hb', ln=None, **ctx):
"""Return records using Jinja template."""
from flask import request
from invenio_base.i18n import wash_language
from .registry import export_formats
of = of.lower()
jrec = request.values.get('jrec', ctx.get('jrec', 1), type=int)
rg = request.values.get('rg', ctx.get('rg', 10), type=int)
ln = ln or wash_language(request.values.get('ln', cfg['CFG_SITE_LANG']))
ot = (request.values.get('ot', ctx.get('ot')) or '').split(',')
if jrec > records:
jrec = rg * (records // rg) + 1
context = dict(
of=of,
jrec=jrec,
rg=rg,
ln=ln,
ot=ot,
facets={},
time=time,
records=records,
export_formats=export_formats,
format_record=format_record,
**TEMPLATE_CONTEXT_FUNCTIONS_CACHE.template_context_functions)
context.update(ctx)
return render_template_to_string([
'format/records/%s.tpl' % of,
'format/records/%s.tpl' % of[0],
'format/records/%s.tpl' %
get_output_format_content_type(of).replace('/', '_')
], **context)
def format_records(records, of='hb', ln=None, **ctx):
"""Return records using Jinja template."""
from flask import request
from invenio_base.i18n import wash_language
from .registry import export_formats
of = of.lower()
jrec = request.values.get('jrec', ctx.get('jrec', 1), type=int)
rg = request.values.get('rg', ctx.get('rg', 10), type=int)
ln = ln or wash_language(request.values.get('ln', cfg['CFG_SITE_LANG']))
ot = (request.values.get('ot', ctx.get('ot')) or '').split(',')
if jrec > records:
jrec = rg * (records // rg) + 1
context = dict(
of=of, jrec=jrec, rg=rg, ln=ln, ot=ot,
facets={},
time=time,
records=records,
export_formats=export_formats,
format_record=format_record,
**TEMPLATE_CONTEXT_FUNCTIONS_CACHE.template_context_functions
)
context.update(ctx)
return render_template_to_string(
['format/records/%s.tpl' % of,
'format/records/%s.tpl' % of[0],
'format/records/%s.tpl' % get_output_format_content_type(of).
replace('/', '_')],
**context)
def get_locale():
"""Compute the language needed to return the answer to the client."""
from invenio_base.i18n import wash_language
required_ln = None
passed_ln = request.values.get('ln', type=str)
if passed_ln:
# If ln is specified explictly as a GET or POST argument
# let's take it!
required_ln = wash_language(passed_ln)
if passed_ln != required_ln:
# But only if it was a valid language
required_ln = None
if required_ln is None and 'ln' not in session:
# If there is no language saved into the session...
user_language = current_user.get('language')
if user_language:
# ... and the user is logged in, we try to take it from its
# settings.
required_ln = user_language
else:
# Otherwise we try to guess it from its request headers
for value, quality in request.accept_languages:
value = str(value)
ln = wash_language(value)
if ln == value or ln[:2] == value[:2]:
required_ln = ln
break
else:
# Too bad! We stick to the default :-)
required_ln = current_app.config.get('CFG_SITE_LANG')
elif required_ln is None:
required_ln = session.get('ln')
assert required_ln is not None
if required_ln != session.get('ln',
current_app.config.get('CFG_SITE_LANG')):
session['ln'] = required_ln
return required_ln
def get_locale():
"""Compute the language needed to return the answer to the client."""
from invenio_base.i18n import wash_language
required_ln = None
passed_ln = request.values.get('ln', type=str)
if passed_ln:
# If ln is specified explictly as a GET or POST argument
# let's take it!
required_ln = wash_language(passed_ln)
if passed_ln != required_ln:
# But only if it was a valid language
required_ln = None
if required_ln is None and 'ln' not in session:
# If there is no language saved into the session...
user_language = current_user.get('language')
if user_language:
# ... and the user is logged in, we try to take it from its
# settings.
required_ln = user_language
else:
# Otherwise we try to guess it from its request headers
for value, quality in request.accept_languages:
value = str(value)
ln = wash_language(value)
if ln == value or ln[:2] == value[:2]:
required_ln = ln
break
else:
# Too bad! We stick to the default :-)
required_ln = current_app.config.get('CFG_SITE_LANG')
elif required_ln is None:
required_ln = session.get('ln')
assert required_ln is not None
if required_ln != session.get('ln',
current_app.config.get('CFG_SITE_LANG')):
session['ln'] = required_ln
return required_ln
def test_wash_invalid_language(self):
"""messages - washing invalid language code"""
self.assertEqual(messages.wash_language('python'),
cfg['CFG_SITE_LANG'])
def test_wash_invalid_language(self):
"""messages - washing invalid language code"""
self.assertEqual(messages.wash_language('python'),
cfg['CFG_SITE_LANG'])
def wash_urlargd(form, content):
"""
Wash the complete form based on the specification in
content. Content is a dictionary containing the field names as a
key, and a tuple (type, default) as value.
'type' can be list, str, invenio.legacy.wsgi.utils.StringField, int, tuple, or
invenio.legacy.wsgi.utils.Field (for
file uploads).
The specification automatically includes the 'ln' field, which is
common to all queries.
Arguments that are not defined in 'content' are discarded.
Note that in case {list,tuple} were asked for, we assume that
{list,tuple} of strings is to be returned. Therefore beware when
you want to use wash_urlargd() for multiple file upload forms.
@Return: argd dictionary that can be used for passing function
parameters by keywords.
"""
result = {}
content['ln'] = (str, '')
for k, (dst_type, default) in content.items():
try:
value = form[k]
except KeyError:
result[k] = default
continue
# FIXES problems with unicode arguments from Flask
if isinstance(value, unicode):
value = value.encode('utf-8')
src_type = type(value)
# First, handle the case where we want all the results. In
# this case, we need to ensure all the elements are strings,
# and not Field instances.
if src_type in (list, tuple):
if dst_type is list:
result[k] = [str(x) for x in value]
continue
if dst_type is tuple:
result[k] = tuple([str(x) for x in value])
continue
# in all the other cases, we are only interested in the
# first value.
value = value[0]
# Maybe we already have what is expected? Then don't change
# anything.
if isinstance(value, dst_type):
result[k] = value
continue
# Since we got here, 'value' is sure to be a single symbol,
# not a list kind of structure anymore.
if dst_type in (str, int):
try:
result[k] = dst_type(value)
except Exception:
result[k] = default
elif dst_type is tuple:
result[k] = (str(value), )
elif dst_type is list:
result[k] = [str(value)]
else:
raise ValueError(
'cannot cast form value %s of type %r into type %r' %
(value, src_type, dst_type))
result['ln'] = wash_language(result['ln'])
return result
