Changeset 3017

Timestamp:

06/21/08 13:25:07 (5 years ago)

Author:

nopper

Message:

New plugin infrastructure with setup from distutils

Files:

• branch/UmitPlugins/umitPlugin/Containers.py (modified) (5 diffs)

branch/UmitPlugins/umitPlugin/Containers.py

@@ -31 +31 @@
 from xml.dom.minidom import parseString, getDOMImplementation
 from tempfile import mkstemp
+
+# For setup functionality
+from distutils.dist import Distribution
+from distutils.core import setup as dist_setup
+from distutils.command.install import install as installcmd
+from distutils.command.install_lib import install_lib as install_libcmd
+from distutils.command.install_egg_info import install_egg_info as install_egginfocmd
+
+# For removing directory trees we need
+import shutil
 
 # FIXME: add others fields
@@ -156 +166 @@
 
 class PluginWriter(object):
-    def __init__(self, out_file, fields):
+    def __init__(self, **fields):
         # Set to None and filter out the unused fields
         
@@ -170 +180 @@
         
         dirs = {
-            'sources' : '*',
-            'dist' : '*'
+            'bin'  : '*',
+            'data' : '*',
+            'lib'  : '*'
         }
         
-        self.file = ZipFile(out_file, "w")
+        self.file = ZipFile(fields['output'], "w")
         
         for i in dirs:
-            self.dir_foreach(i, dirs[i])
+            self.dir_foreach(os.path.join('output', i), dirs[i])
         
         self.file.writestr("Manifest.xml", self.create_manifest())
         self.file.close()
         
-        print "Plugin %s created." % out_file
-        print ""
+        print ">> Plugin %s created." % fields['output']
     
     def dir_foreach(self, dir, pattern):
@@ -199 +209 @@
                 
                 self.file.write(os.path.join(path, file),
-                                os.path.join(path, file))
+                                os.path.join(os.path.basename(path), file))
     
     def create_manifest(self):
@@ -217 +227 @@
         return doc.toxml()
 
-def setup(name, fields):
+
+#
+# distutils related class
+
+class PlugLibInstaller(install_libcmd):
+    def finalize_options(self):
+        self.install_dir = 'output/lib'
+        install_libcmd.finalize_options(self)
+
+class PlugInstaller(installcmd):
+    def finalize_options(self):
+        self.home = 'output'
+        installcmd.finalize_options(self)
+
+    def run(self):
+        installcmd.run(self)
+
+class PlugEggInstaller(install_egginfocmd):
+    def run(self):
+        pass
+
+class PlugDistribution(Distribution):
+    def __init__(self, *attrs):
+        Distribution.__init__(self, *attrs)
+        self.cmdclass['install'] = PlugInstaller
+        self.cmdclass['install_lib'] = PlugLibInstaller
+        self.cmdclass['install_egg_info'] = PlugEggInstaller
+
+
+def setup(**attrs):
     "Called to create a plugin like the dist-tools setup function"
     
-    # TODO: add the compile flag
-    PluginWriter("%s.ump" % name, fields)
-
-if __name__ == "__main__":
-    PluginWriter("test/out.ump",
-        {
-            "start-file" : "main.py",
-            "url": "http://test.org",
-            "unz": "non-valid-entry"
-        }
-    )
+    print ">> Running setup()"
+    attrs['distclass'] = PlugDistribution
+    dist_setup(**attrs)
+
+    print ">> Creating plugin"
+    PluginWriter(**attrs)
+
+    print ">> Cleaning up"
+    shutil.rmtree('build')
+    shutil.rmtree('output')