add schema.code back

Author: dimitri-yatsenko

datajoint/diagram.py

@@ -178,9 +178,9 @@ def is_part(part, master):
             return self
 
         def topological_sort(self):
-            """:return:  list of nodes in topological order"""
+            """:return:  list of nodes in lexcigraphical topological order"""
             return list(
-                nx.algorithms.dag.topological_sort(
+                nx.algorithms.dag.lexicographical_topological_sort(
                     nx.DiGraph(self).subgraph(self.nodes_to_show)
                 )
             )

datajoint/schemas.py

@@ -2,16 +2,18 @@
 import logging
 import inspect
 import re
+import collections
+import itertools
 from .connection import conn
-from .diagram import Diagram
+from .diagram import Diagram, _get_tier
 from .settings import config
 from .errors import DataJointError, AccessError
 from .jobs import JobTable
 from .external import ExternalMapping
 from .heading import Heading
 from .utils import user_choice, to_camel_case
 from .user_tables import Part, Computed, Imported, Manual, Lookup
-from .table import lookup_class_name, Log
+from .table import lookup_class_name, Log, FreeTable
 import types
 
 logger = logging.getLogger(__name__.split(".")[0])
@@ -399,6 +401,78 @@ def jobs(self):
             self._jobs = JobTable(self.connection, self.database)
         return self._jobs
 
+    @property
+    def code(self):
+        self._assert_exists()
+        return self.save()
+
+    def save(self, python_filename=None):
+        """
+        Generate the code for a module that recreates the schema.
+        This method is in preparation for a future release and is not officially supported.
+
+        :return: a string containing the body of a complete Python module defining this schema.
+        """
+        self._assert_exists()
+        module_count = itertools.count()
+        # add virtual modules for referenced modules with names vmod0, vmod1, ...
+        module_lookup = collections.defaultdict(
+            lambda: "vmod" + str(next(module_count))
+        )
+        db = self.database
+
+        def make_class_definition(table):
+            tier = _get_tier(table).__name__
+            class_name = table.split(".")[1].strip("`")
+            indent = ""
+            if tier == "Part":
+                class_name = class_name.split("__")[-1]
+                indent += "    "
+            class_name = to_camel_case(class_name)
+
+            def replace(s):
+                d, tabs = s.group(1), s.group(2)
+                return ("" if d == db else (module_lookup[d] + ".")) + ".".join(
+                    to_camel_case(tab) for tab in tabs.lstrip("__").split("__")
+                )
+
+            return ("" if tier == "Part" else "\n@schema\n") + (
+                "{indent}class {class_name}(dj.{tier}):\n"
+                '{indent}    definition = """\n'
+                '{indent}    {defi}"""'
+            ).format(
+                class_name=class_name,
+                indent=indent,
+                tier=tier,
+                defi=re.sub(
+                    r"`([^`]+)`.`([^`]+)`",
+                    replace,
+                    FreeTable(self.connection, table).describe(),
+                ).replace("\n", "\n    " + indent),
+            )
+
+        diagram = Diagram(self)
+        body = "\n\n".join(
+            make_class_definition(table) for table in diagram.topological_sort()
+        )
+        python_code = "\n\n".join(
+            (
+                '"""This module was auto-generated by datajoint from an existing schema"""',
+                "import datajoint as dj\n\nschema = dj.Schema('{db}')".format(db=db),
+                "\n".join(
+                    "{module} = dj.VirtualModule('{module}', '{schema_name}')".format(
+                        module=v, schema_name=k
+                    )
+                    for k, v in module_lookup.items()
+                ),
+                body,
+            )
+        )
+        if python_filename is None:
+            return python_code
+        with open(python_filename, "wt") as f:
+            f.write(python_code)
+
     def list_tables(self):
         """
         Return a list of all tables in the schema except tables with ~ in first character such