1 files changed, 96 insertions, 0 deletions
diff --git a/bin/tools/upgrade.py b/bin/tools/upgrade.py
new file mode 100644
index 00000000000..ef377e83c85
--- /dev/null
+++ b/bin/tools/upgrade.py
@@ -0,0 +1,96 @@
+##############################################################################
+#
+# Copyright (c) 2004 TINY SPRL. (http://tiny.be) All Rights Reserved.
+#
+# $Id: config.py 996 2005-07-22 10:40:10Z nicoe $
+#
+# WARNING: This program as such is intended to be used by professional
+# programmers who take the whole responsability of assessing all potential
+# consequences resulting from its eventual inadequacies and bugs
+# End users who are looking for a ready-to-use solution with commercial
+# garantees and support are strongly adviced to contract a Free Software
+# Service Company
+#
+# This program is Free Software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public License
+# as published by the Free Software Foundation; either version 2
+# of the License, or (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
+#
+##############################################################################
+
+import tarfile
+import re
+import urllib2
+import os
+import shutil
+import tools
+
+
+# remove an existing version of modules if it exist
+def remove(name):
+	adp = tools.config['addons_path']
+	addons = os.listdir(adp)
+	if name in addons:
+		try:
+			shutil.rmtree(os.path.join(adp, name))
+		except:
+			print "Unable to remove module %s !" % name
+
+def install(name, url):
+	tar = tarfile.open(mode="r|gz", fileobj=urllib2.urlopen(url))
+	for tarinfo in tar:
+		tar.extract(tarinfo, tools.config['addons_path'])
+
+def upgrade():
+	import pooler
+	cr = pooler.db.cursor()
+
+	toinit = []
+	toupdate = []
+
+#	print 'Check for correct rights (create and unlink on addons)...'
+	# todo: touch addons/test.txt
+	# todo: rm addons/test.txt
+
+	print 'Check for modules to remove...'
+	cr.execute('select id,name,url from ir_module_module where state=%s', ('to remove',))
+	for module_id,name,url in cr.fetchall():
+		print '\tremoving module %s' % name
+		remove(name)
+		cr.execute('update ir_module_module set state=%s where id=%d', ('uninstalled', module_id))
+		cr.commit()
+
+	print 'Check for modules to upgrade...'
+	cr.execute('select id,name,url from ir_module_module where state=%s', ('to upgrade',))
+	for module_id,name,url in cr.fetchall():
+		print '\tupgrading module %s' % name
+		remove(name)
+		install(name, url)
+		cr.execute('update ir_module_module set state=%s where id=%d', ('installed', module_id))
+		cr.commit()
+		toupdate.append(name)
+
+	print 'Check for modules to install...'
+	cr.execute('select id,name,url from ir_module_module where state=%s', ('to install',))
+	for module_id,name,url in cr.fetchall():
+		print '\tinstalling module %s' % name
+		install(name, url)
+		cr.execute('update ir_module_module set state=%s where id=%d', ('installed', module_id))
+		cr.commit()
+		toinit.append(name)
+
+	print 'Initializing all datas...'
+
+	cr.commit()
+	cr.close()
+	return (toinit, toupdate)
+