Appropriate context

Author: Peyman-N

pipeline/src/module_template.py.txt

@@ -16,7 +16,7 @@ class {{ class_name }}({{ base_class }}):
     """
     type_ = "{{ openminds_type }}"
     context = {
-        "@vocab": "https://openminds.ebrains.eu/vocab/"
+        "@vocab": "https://openminds.om-i.org/vocab/"
     }
     schema_version = "{{ schema_version }}"
 
@@ -61,4 +61,4 @@ class {{ class_name }}({{ base_class }}):
         {{key}}={{value}},
     {%- endif %}
     {% endfor -%}
-){% endfor %}
+){% endfor %}

pipeline/src/module_template_old.py.txt

@@ -0,0 +1,64 @@
+"""
+{{docstring}}
+"""
+
+# this file was auto-generated!
+
+{{preamble}}
+
+from openminds.base import {{ base_class }}
+from openminds.properties import Property
+
+
+class {{ class_name }}({{ base_class }}):
+    """
+    {{ docstring }}
+    """
+    type_ = "{{ openminds_type }}"
+    context = {
+        "@vocab": "https://openminds.ebrains.eu/vocab/"
+    }
+    schema_version = "{{ schema_version }}"
+
+    properties = [
+        {% for property in properties -%}
+        Property(
+            "{{property.name}}",
+            {{property.type}},
+            "{{property.iri}}",
+            {%- if property.allow_multiple %}
+            multiple={{property.allow_multiple}},
+            unique_items={{property.unique_items}},
+            {% if property.min_items -%}min_items={{property.min_items}},{% endif %}
+            {% if property.max_items -%}max_items={{property.max_items}},{% endif %}
+            {% endif -%}
+            {% if property.type == "str" -%}
+            formatting="{{property.formatting}}",
+            {% if property.multiline -%}multiline={{property.multiline}},{% endif %}
+            {% endif %}
+            {%- if property.required -%}required={{property.required}},{% endif %}
+            description="{{property.description}}",
+            instructions="{{property.instructions}}"
+        ),
+        {% endfor %}
+    ]
+
+    def __init__(self{%- if base_class == "LinkedMetadata" %}, id=None{%- endif -%}{%- for property in properties -%}, {{property.name}}=None{%- endfor -%}):
+        return super().__init__({%- if base_class == "LinkedMetadata" %}id=id, {%- endif -%}{%- for property in properties -%}{{property.name}}={{property.name}}, {%- endfor -%})
+
+{{ additional_methods }}
+
+{% for instance_name, instance in instances.items() %}
+{{ class_name }}.{{ instance_name }} = {{ class_name }}(
+    {% for key, value in instance.items() -%}
+    {% if value is string -%}
+        {% if value.startswith('http') and key != 'id' -%}
+            {{key}}=IRI("{{value}}"),
+        {%- else -%}
+            {{key}}="{{value}}",
+        {%- endif %}
+    {%- else -%}
+        {{key}}={{value}},
+    {%- endif %}
+    {% endfor -%}
+){% endfor %}

pipeline/translator.py

@@ -57,14 +57,17 @@ class PythonBuilder(object):
 
     def __init__(self, schema_file_path: str, root_path: str, instances: Optional[dict] = None,
                  additional_methods: Optional[dict] = None):
-        self.template_name = "src/module_template.py.txt"
         self.env = Environment(
             loader=FileSystemLoader(os.path.dirname(os.path.realpath(__file__))), autoescape=select_autoescape()
         )
         _relative_path_without_extension = (
             schema_file_path[len(root_path) + 1 :].replace(".schema.omi.json", "").split("/")
         )
         self.version = _relative_path_without_extension[0]
+        if self.version in ["v3.0" , "v2.0" , "v1.0"]:
+            self.template_name = "src/module_template_old.py.txt"
+        else:
+            self.template_name = "src/module_template.py.txt"
         self.relative_path_without_extension = [
             generate_python_name(part) for part in _relative_path_without_extension[1:]
         ]