blob: 4b0af166b901cbf6b688cf9aa7e1f07dd670f5c5 [file] [log] [blame]
from base import *
import pdb
def django_content_type_string(xptags):
# Check possibility of KeyError in caller
content_type = xptags['content_type']
try:
content_type = eval(content_type)
except:
pass
if (content_type=='url'):
return 'URLField'
if (content_type=='date'):
return 'DateTimeField'
elif (content_type=='ip'):
return 'GenericIPAddressField'
elif (content_type=='stripped' or content_type=='"stripped"'):
return 'StrippedCharField'
else:
raise Exception('Unknown Type: %s'%content_type)
def django_string_type(xptags):
try:
max_length = eval(xptags['max_length'])
except:
max_length = 1024 * 1024
if ('content_type' in xptags):
return django_content_type_string(xptags)
elif (max_length<1024*1024):
return 'CharField'
else:
return 'TextField'
def xproto_django_type(xptype, xptags):
if (xptype=='string'):
return django_string_type(xptags)
elif (xptype=='float'):
return 'FloatField'
elif (xptype=='bool'):
return 'BooleanField'
elif (xptype=='uint32'):
return 'IntegerField'
elif (xptype=='int32'):
return 'IntegerField'
elif (xptype=='int64'):
return 'BigIntegerField'
else:
raise Exception('Unknown Type: %s'%xptype)
def xproto_django_link_type(f):
if (f['link_type']=='manytoone'):
return 'ForeignKey'
elif (f['link_type']=='onetoone'):
return 'OneToOneField'
elif (f['link_type']=='manytomany'):
if (f['dst_port']):
return 'ManyToManyField'
else:
return 'GenericRelation'
def map_xproto_to_django(f):
allowed_keys=['help_text','default','max_length','modifier','blank','choices','db_index','null','editable','on_delete','verbose_name', 'auto_now_add']
m = {'modifier':{'optional':True, 'required':False, '_target':'null'}}
out = {}
for k,v in f['options'].items():
if (k in allowed_keys):
try:
kv2 = m[k]
out[kv2['_target']] = kv2[v]
except:
out[k] = v
return out
def xproto_django_link_options_str(field, dport=None):
output_dict = map_xproto_to_django(field)
if (dport and (dport=='+' or '+' not in dport)):
output_dict['related_name'] = '%r'%dport
try:
if field['through']:
d = {}
if isinstance(field['through'], str):
split = field['through'].rsplit('.',1)
d['name'] = split[-1]
if len(split)==2:
d['package'] = split[0]
d['fqn'] = 'package' + '.' + d['name']
else:
d['fqn'] = d['name']
d['package'] = ''
else:
d = field['through']
if not d['name'].endswith('_'+field['name']):
output_dict['through'] = '%r'%d['fqn']
except KeyError:
pass
return format_options_string(output_dict)
def format_options_string(d):
if (not d):
return ''
else:
lst = []
for k,v in d.items():
if (type(v)==str and k=='default' and v.endswith('()"')):
lst.append('%s = %s'%(k,v[1:-3]))
elif (type(v)==str and v.startswith('"')):
try:
tup = eval(v[1:-1])
if (type(tup)==tuple):
lst.append('%s = %r'%(k,tup))
else:
lst.append('%s = %s'%(k,v))
except:
lst.append('%s = %s'%(k,v))
elif (type(v)==bool):
lst.append('%s = %r'%(k,bool(v)))
else:
try:
lst.append('%s = %r'%(k,int(v)))
except ValueError:
lst.append('%s = %s'%(k,v))
return ', '.join(lst)
def xproto_django_options_str(field, dport=None):
output_dict = map_xproto_to_django(field)
if (dport=='_'):
dport = '+'
if (dport and (dport=='+' or '+' not in dport)):
output_dict['related_name'] = '%r'%dport
return format_options_string(output_dict)
def xproto_validations(options):
try:
return [map(str.strip, validation.split(':')) for validation in unquote(options['validators']).split(',')]
except KeyError:
return []