Merge pull request #5014 from dczech/feature/decouple-bigautofield

Include URL path regexs in API schemas where valid.
This commit is contained in:
Tom Christie 2017-03-22 09:37:07 +00:00 committed by GitHub
commit 8e9cb1e614

View File

@ -526,6 +526,7 @@ class SchemaGenerator(object):
title = ''
description = ''
schema_cls = coreschema.String
kwargs = {}
if model is not None:
# Attempt to infer a field description if possible.
try:
@ -541,14 +542,16 @@ class SchemaGenerator(object):
elif model_field is not None and model_field.primary_key:
description = get_pk_description(model, model_field)
if isinstance(model_field, models.AutoField):
if hasattr(view, 'lookup_value_regex') and view.lookup_field == variable:
kwargs['pattern'] = view.lookup_value_regex
elif isinstance(model_field, models.AutoField):
schema_cls = coreschema.Integer
field = coreapi.Field(
name=variable,
location='path',
required=True,
schema=schema_cls(title=title, description=description)
schema=schema_cls(title=title, description=description, **kwargs)
)
fields.append(field)