etwist/twctl.py
changeset 9372 e87a808a4c32
parent 7619 0d0344fd5231
child 10237 1e030b1a5622
--- a/etwist/twctl.py	Thu Dec 12 16:18:58 2013 +0100
+++ b/etwist/twctl.py	Thu Dec 12 18:25:45 2013 +0100
@@ -22,7 +22,7 @@
 from logilab.common.shellutils import rm
 
 from cubicweb.toolsutils import CommandHandler
-from cubicweb.web.webctl import WebCreateHandler
+from cubicweb.web.webctl import WebCreateHandler, WebUpgradeHandler
 
 # trigger configuration registration
 import cubicweb.etwist.twconfig # pylint: disable=W0611
@@ -48,6 +48,9 @@
     def poststop(self):
         pass
 
+class TWUpgradeHandler(WebUpgradeHandler):
+    cfgname = 'twisted'
+
 
 try:
     from cubicweb.server import serverctl
@@ -73,5 +76,8 @@
         cfgname = 'all-in-one'
         subcommand = 'cubicweb-twisted'
 
+    class AllInOneUpgradeHandler(TWUpgradeHandler):
+        cfgname = 'all-in-one'
+
 except ImportError:
     pass