From 6a18f8bd17ece8c51eb24ac414e69112b7e71e9c Mon Sep 17 00:00:00 2001
From: Antoine Nguyen <tonio@ngyn.org>
Date: Thu, 2 Feb 2012 11:12:46 +0100
Subject: [PATCH] Works better with appriopriate template :p

see #201
---
 extensions/stats/grapher.py             |  3 ++-
 templates/admin/view_domain_admins.html |  8 +++++++
 tools/pfxadmin_migrate/migrate.py       | 30 +++++++++++--------------
 3 files changed, 23 insertions(+), 18 deletions(-)
 create mode 100644 templates/admin/view_domain_admins.html

diff --git a/extensions/stats/grapher.py b/extensions/stats/grapher.py
index ad52da760..4237ed7d9 100644
--- a/extensions/stats/grapher.py
+++ b/extensions/stats/grapher.py
@@ -3,7 +3,6 @@
 import os
 import time
 import datetime
-import rrdtool
 from django.utils.translation import ugettext as _
 from modoboa.lib import parameters
 
@@ -77,6 +76,8 @@ class Grapher(object):
         self.img_rootdir = parameters.get_admin("IMG_ROOTDIR")
 
     def process(self, target, suffix, start, end, tpl=traffic_avg_template):
+        import rrdtool
+
         rrdfile = "%s/%s.rrd" % (self.rrd_rootdir, target)
         if not os.path.exists(rrdfile):
             return
diff --git a/templates/admin/view_domain_admins.html b/templates/admin/view_domain_admins.html
new file mode 100644
index 000000000..403011153
--- /dev/null
+++ b/templates/admin/view_domain_admins.html
@@ -0,0 +1,8 @@
+{% extends "layout_simple.html" %}
+{% load i18n %}
+{% block content %}
+<div class="title"><h1>{% trans "Administrators of" %} {{ domain.name }}</h1></div>
+<select id="domainadmins" multiple="multiple">
+{% for owner in domain.owners.all %}<option value="{{ owner.user.id }}">{{ owner.user.username }}</option>{% endfor %}
+</select>
+{% endblock %}
diff --git a/tools/pfxadmin_migrate/migrate.py b/tools/pfxadmin_migrate/migrate.py
index 0bce33811..499818a52 100755
--- a/tools/pfxadmin_migrate/migrate.py
+++ b/tools/pfxadmin_migrate/migrate.py
@@ -115,9 +115,7 @@ def migrate_mailboxes(domain, options):
     old_mboxes = pf_models.Mailbox.objects.using(options._from).filter(domain=domain.name)
     for old_mb in old_mboxes:
         new_mb = md_models.Mailbox()
-        new_mb.name = old_mb.name
         new_mb.address = old_mb.local_part
-        new_mb.password = old_mb.password
         new_mb.domain = domain
         new_mb.dates = migrate_dates(old_mb)
         if old_mb.quota:
@@ -132,7 +130,7 @@ def migrate_mailboxes(domain, options):
                 print "Error: cannot rename mailbox directory\n%s" % output
                 sys.exit(1)
 
-        new_mb.save(using=options.to)
+        new_mb.save(name=old_mb.name, password=old_mb.password, using=options.to)
         
 
 def migrate_domain(old_dom, options):
@@ -161,29 +159,27 @@ def migrate_admins(options):
             print "Warning: skipping useless admin %s" % (old_admin.username)
             continue
         try:
-            mb = md_models.Mailbox.objects.using(options.to).get(user__username=old_admin.username)
-        except md_models.Mailbox.DoesNotExist:
+            user = md_models.User.objects.using(options.to).get(username=old_admin.username)
+        except md_models.User.DoesNotExist:
             try:
                 domain = md_models.Domain.objects.using(options.to).get(name=domname)
             except md_models.Domain.DoesNotExist:
                 print "Warning: skipping domain admin %s, domain not found" \
                     % old_admin.username
                 continue
-            mb = md_models.Mailbox()
-            mb.address = local_part
-            mb.domain = domain
-            mb.name = local_part
-            mb.enabled = old_admin.active
-            mb.password = old_admin.password
-            mb.dates = migrate_dates(old_admin)
-            mb.save(using=options.to)            
+            user = md_models.User()
+            user.username = old_admin.username
+            user.email = old_admin.username
+            user.password = old_admin.password
+            user.is_active = old_admin.active
+            user.save(using=options.to)
 
-        mb.user.date_joined = old_admin.modified
+        user.date_joined = old_admin.modified
         if creds.domain == "ALL":
-            mb.user.is_superuser = True
+            user.is_superuser = True
         else:
-            mb.user.groups.add(dagroup)
-        mb.save(using=options.to)
+            user.groups.add(dagroup)
+        user.save(using=options.to)
 
 def do_migration(options):
     pf_domains = pf_models.Domain.objects.using(options._from).all()