blob: 67e25c5e89903899f3d61adc5818658b66b26eaa [file] [log] [blame]
Sapan Bhatiad022aeb2017-06-07 15:49:55 +02001{% for m in proto.messages %}{% if not m.options.skip_django -%}
Sapan Bhatia4e80a262017-05-19 23:10:51 +02002{% if file_exists(xproto_base_name(m.name)|lower+'_header.py') -%}from {{xproto_base_name(m.name)|lower }}_header import *{%- else -%}from header import *{% endif %}
Sapan Bhatiac4f803f2017-04-21 11:50:39 +02003{% if file_exists(xproto_base_name(m.name)|lower+'_top.py') -%}{{ include_file(xproto_base_name(m.name)|lower+'_top.py') }} {% endif %}
Matteo Scandolo67654fa2017-06-09 09:33:17 -07004
Sapan Bhatiaff1b8fa2017-04-10 19:44:38 -07005{%- for l in m.links %}
Sapan Bhatiac4f803f2017-04-21 11:50:39 +02006
Sapan Bhatia3cfdf632017-06-08 05:14:03 +02007{% if l.peer.name != m.name %}
Matteo Scandolo67654fa2017-06-09 09:33:17 -07008from core.models.{{ l.peer.name | lower }} import {{ l.peer.name }}
Sapan Bhatiac4f803f2017-04-21 11:50:39 +02009{% endif %}
10
Sapan Bhatiaff1b8fa2017-04-10 19:44:38 -070011{%- endfor %}
Sapan Bhatiac4f803f2017-04-21 11:50:39 +020012{% for b in m.bases %}
Sapan Bhatia3cfdf632017-06-08 05:14:03 +020013{% if b.name!='XOSBase' and 'Mixin' not in b.name %}
14from core.models.{{b.name | lower}} import {{ b.name }}
Sapan Bhatiac4f803f2017-04-21 11:50:39 +020015{% endif %}
16{% endfor %}
17
Sapan Bhatiaff1b8fa2017-04-10 19:44:38 -070018
Sapan Bhatia4e80a262017-05-19 23:10:51 +020019class {{ m.name }}{{ xproto_base_def(m.name, m.bases) }}:
Sapan Bhatia2941a052017-07-10 15:10:03 -040020 plural_name = "{{ xproto_pluralize(m) }}"
21
Sapan Bhatiaff1b8fa2017-04-10 19:44:38 -070022 # Primitive Fields (Not Relations)
Sapan Bhatiac4f803f2017-04-21 11:50:39 +020023 {% for f in m.fields %}
24 {%- if not f.link -%}
Sapan Bhatiaff1b8fa2017-04-10 19:44:38 -070025 {{ f.name }} = {{ xproto_django_type(f.type, f.options) }}( {{ xproto_django_options_str(f) }} )
Sapan Bhatiac4f803f2017-04-21 11:50:39 +020026 {% endif %}
Sapan Bhatiaff1b8fa2017-04-10 19:44:38 -070027 {%- endfor %}
28
29 # Relations
Sapan Bhatia3cfdf632017-06-08 05:14:03 +020030 {% for l in m.links %}{% set peer_name=l.peer.name %}
31 {{ l.src_port }} = {{ xproto_django_link_type(l) }}( {%- if peer_name==m.name -%}'self'{%- else -%}{{ peer_name }} {%- endif -%}, {{ xproto_django_link_options_str(l, l.dst_port ) }} )
Sapan Bhatiaff1b8fa2017-04-10 19:44:38 -070032 {%- endfor %}
33
Sapan Bhatiaf7934b52017-06-12 05:04:23 -070034 # Meta
35 {%- set uniques = xproto_field_graph_components(m.fields) %}
36 {%- if uniques %}
37 class Meta:
38 unique_together = {{ xproto_tuplify(uniques) }}
39 {%- endif %}
Sapan Bhatiac4f803f2017-04-21 11:50:39 +020040 {% if file_exists(m.name|lower + '_model.py') -%}{{ include_file(m.name|lower + '_model.py') | indent(width=2)}}{%- endif %}
41 pass
Sapan Bhatiaff1b8fa2017-04-10 19:44:38 -070042
Sapan Bhatiac4f803f2017-04-21 11:50:39 +020043{% if file_exists(xproto_base_name(m.name)|lower+'_bottom.py') -%}{{ include_file(xproto_base_name(m.name)|lower+'_bottom.py') }}{% endif %}
Sapan Bhatia2941a052017-07-10 15:10:03 -040044{% endif %}{% endfor %}