dak/import_users_from_passwd.py

   1 #!/usr/bin/env python
   2
   3 # Sync PostgreSQL users with system users
   4 # Copyright (C) 2001, 2002  James Troup <james@nocrew.org>
   5 # $Id: julia,v 1.9 2003-01-02 18:12:50 troup Exp $
   6
   7 # This program is free software; you can redistribute it and/or modify
   8 # it under the terms of the GNU General Public License as published by
   9 # the Free Software Foundation; either version 2 of the License, or
  10 # (at your option) any later version.
  11
  12 # This program is distributed in the hope that it will be useful,
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 # GNU General Public License for more details.
  16
  17 # You should have received a copy of the GNU General Public License
  18 # along with this program; if not, write to the Free Software
  19 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  20
  21 ################################################################################
  22
  23 # <aj> ARRRGGGHHH
  24 # <aj> what's wrong with me!?!?!?
  25 # <aj> i was just nice to some mormon doorknockers!!!
  26 # <Omnic> AJ?!?!
  27 # <aj> i know!!!!!
  28 # <Omnic> I'm gonna have to kick your ass when you come over
  29 # <Culus> aj: GET THE HELL OUT OF THE CABAL! :P
  30
  31 ################################################################################
  32
  33 import pg, pwd, sys;
  34 import utils;
  35 import apt_pkg;
  36
  37 ################################################################################
  38
  39 Cnf = None;
  40 projectB = None;
  41 ################################################################################
  42
  43 def usage (exit_code=0):
  44     print """Usage: julia [OPTION]...
  45 Sync PostgreSQL's users with system users.
  46
  47   -h, --help                 show this help and exit
  48   -n, --no-action            don't do anything
  49   -q, --quiet                be quiet about what is being done
  50   -v, --verbose              explain what is being done"""
  51     sys.exit(exit_code)
  52
  53 ################################################################################
  54
  55 def main ():
  56     global Cnf, projectB;
  57
  58     Cnf = utils.get_conf()
  59
  60     Arguments = [('n', "no-action", "Julia::Options::No-Action"),
  61                  ('q', "quiet", "Julia::Options::Quiet"),
  62                  ('v', "verbose", "Julia::Options::Verbose"),
  63                  ('h', "help", "Julia::Options::Help")];
  64     for i in [ "no-action", "quiet", "verbose", "help" ]:
  65         if not Cnf.has_key("Julia::Options::%s" % (i)):
  66             Cnf["Julia::Options::%s" % (i)] = "";
  67
  68     arguments = apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv);
  69     Options = Cnf.SubTree("Julia::Options")
  70
  71     if Options["Help"]:
  72         usage();
  73     elif arguments:
  74         utils.warn("julia takes no non-option arguments.");
  75         usage(1);
  76
  77     projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"], int(Cnf["DB::Port"]));
  78     valid_gid = int(Cnf.get("Julia::ValidGID",""));
  79
  80     passwd_unames = {};
  81     for entry in pwd.getpwall():
  82         uname = entry[0];
  83         gid = entry[3];
  84         if valid_gid and gid != valid_gid:
  85             if Options["Verbose"]:
  86                 print "Skipping %s (GID %s != Valid GID %s)." % (uname, gid, valid_gid);
  87             continue;
  88         passwd_unames[uname] = "";
  89
  90     postgres_unames = {};
  91     q = projectB.query("SELECT usename FROM pg_user");
  92     ql = q.getresult();
  93     for i in ql:
  94         uname = i[0];
  95         postgres_unames[uname] = "";
  96
  97     known_postgres_unames = {};
  98     for i in Cnf.get("Julia::KnownPostgres","").split(","):
  99         uname = i.strip();
 100         known_postgres_unames[uname] = "";
 101
 102     keys = postgres_unames.keys()
 103     keys.sort();
 104     for uname in keys:
 105         if not passwd_unames.has_key(uname)and not known_postgres_unames.has_key(uname):
 106             print "W: %s is in Postgres but not the passwd file or list of known Postgres users." % (uname);
 107
 108     keys = passwd_unames.keys()
 109     keys.sort();
 110     for uname in keys:
 111         if not postgres_unames.has_key(uname):
 112             if not Options["Quiet"]:
 113                 print "Creating %s user in Postgres." % (uname);
 114             if not Options["No-Action"]:
 115                 q = projectB.query('CREATE USER "%s"' % (uname));
 116
 117 #######################################################################################
 118
 119 if __name__ == '__main__':
 120     main()
 121