SF.net SVN: postfixadmin:[529] trunk/ADDITIONS/quota_usage.pl

[<Gin...@us...>]

Revision: 529
          http://postfixadmin.svn.sourceforge.net/postfixadmin/?rev=529&view=rev
Author:   GingerDog
Date:     2009-01-18 12:03:14 +0000 (Sun, 18 Jan 2009)

Log Message:
-----------
adding file from Jose Nilton <jni...@gm...> - produces report of quota usage etc

Added Paths:
-----------
    trunk/ADDITIONS/quota_usage.pl

Added: trunk/ADDITIONS/quota_usage.pl
===================================================================
--- trunk/ADDITIONS/quota_usage.pl	                        (rev 0)
+++ trunk/ADDITIONS/quota_usage.pl	2009-01-18 12:03:14 UTC (rev 529)
@@ -0,0 +1,140 @@
+#!/usr/bin/perl
+# vim:ts=4:sw=4:et
+#
+# Contributed to Postfixadmin by Jose Nilton <jni...@gm...> 
+#
+# See also : http://www.russelldare.net/media/perl/dirsizeSource.pdf
+# License: GPL v2.
+
+# Usage:
+# perl quota_usage.pl --list or  
+# perl quota_usage.pl --addmysql for add mysql database postfix 
+#
+# Requires: perl perl-DBD-mysql perl-DBD (may be named differently depending on your platform).
+#           and the 'du' binary in $ENV{'PATH'} (see below).
+#
+# You will need to modify the postfix DATABASE to add a quota_usage column. 
+# Mysql:
+# 	ALTER TABLE mailbox ADD quota_usage INT(11) NOT NULL DEFAULT '0' AFTER modified, 
+# 	ADD quota_usage_date DATE NOT NULL DEFAULT '0000-00-00' AFTER quota_usage;
+# PostgreSQL:
+# 	ALTER TABLE mailbox ADD COLUMN quota_usage INTEGER NOT NULL DEFAULT 0;
+# 	ALTER TABLE mailbox ADD COLUMN quota_usage_date DATE NOT NULL DEFAULT current_date;
+#
+
+use strict;
+use warnings;
+use File::Path;
+use DBI;
+use Getopt::Long;
+
+##EDIT## 
+my $db_host 	= 'localhost';
+my $db_database = 'postfix';
+my $db_user 	= 'postfix';
+my $db_password = '123456';
+my $root_path 	= '/home/vmail';
+# Pg or mysql
+my $db_type     = 'Pg'; 
+##END EDIT##
+
+$ENV{'PATH'} = "/sbin:/bin:/usr/sbin:/usr/bin";
+my($domain_dir, $full_domain_dir, $user_dir, $usage, $email, $sql, $dbh);
+
+GetOptions ('list' => \&list_quota_usage, 'addmysql' => \&insert_to_db);
+
+sub list_quota_usage {
+    opendir(DOMAINDIR, $root_path) or die ("Unable to access directory '$root_path' ($!)");
+
+    foreach $domain_dir (sort readdir DOMAINDIR) {
+        next if $domain_dir =~ /^\./;                # skip dotted dirs
+            $full_domain_dir = "$root_path/$domain_dir"; #print "$full_domain_dir\n";
+
+        opendir(USERDIR, $full_domain_dir) or die ("Unable to access directory '$full_domain_dir' ($!)");
+        foreach $user_dir (sort readdir USERDIR) {
+            next if $user_dir =~ /^\./; # skip dotted dirs
+                $email = "$user_dir\@$domain_dir";
+
+            my $i = `du -0 --summarize $full_domain_dir/$user_dir`;
+            ($usage) = split(" ", $i);
+
+            if ($usage < 100) {
+                $usage = 0;
+            } elsif ($usage < 1000) {
+                $usage = 1;
+            } else {
+                $usage = $usage + 500;
+                $usage = int $usage / 1000;
+            }
+
+            list_out();
+        }
+    }
+    close(DOMAINDIR);
+    close(USERDIR);
+}
+
+
+sub insert_to_db {
+    opendir(DOMAINDIR, $root_path) or die ("Unable to access directory '$root_path' ($!)");
+
+    $dbh = DBI->connect("DBI:$db_type:database=$db_database;host=$db_host", $db_user, $db_password) or die ("cannot connect the database");
+    execSql("UPDATE mailbox set quota_usage = 0");
+
+
+    foreach $domain_dir (sort readdir DOMAINDIR) {
+        next if $domain_dir =~ /^\./; # skip dotted dirs
+            $full_domain_dir = "$root_path/$domain_dir"; #print "$full_domain_dir\n";
+
+        opendir(USERDIR, $full_domain_dir) or die ("Unable to access directory '$full_domain_dir' ($!)");
+        foreach $user_dir (sort readdir USERDIR) {
+            next if $user_dir =~ /^\./; # skip dotted dirs
+                $email = "$user_dir\@$domain_dir";
+
+            my $i = `du -0 --summarize $full_domain_dir/$user_dir`;
+            ($usage) = split(" ", $i);
+
+            if ($usage < 100) {
+                $usage = 0;
+            } elsif ($usage < 1000) {
+                $usage = 1;
+            } else {
+                $usage = $usage + 500;
+                $usage = int $usage / 1000;
+            }
+
+            execSql("UPDATE mailbox set quota_usage = $usage, quota_usage_date = CAST(NOW() AS DATE) WHERE username = '$email'");
+#list_out(); #Debug
+        }
+
+    }
+
+    close(DOMAINDIR);
+    close(USERDIR);
+
+}
+
+
+sub execSql {
+    my $sql = shift;
+    my $ex;
+    $ex = $dbh->do($sql) or die ("error when running $sql");
+}
+
+sub list_out {
+format STDOUT_TOP =
+Report of Quota Used
+--------------------------
+EMAIL                                         QUOTA USED
+------------------------------------------------------------------
+.
+
+
+format = 
+@<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<              @<<<<<<<<<<
+$email,                                                "$usage\bMB"                         
+.
+
+    write;
+}
+


This was sent by the SourceForge.net collaborative development platform, the world's largest Open Source development site.