julia

   1 #!/usr/bin/env python
   2
   3 # Sync PostgreSQL with (LDAP-generated) passwd file
   4 # Copyright (C) 2001  James Troup <james@nocrew.org>
   5 # $Id: julia,v 1.1 2001-09-13 23:55:51 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 # <aj> ARRRGGGHHH
  22 # <aj> what's wrong with me!?!?!?
  23 # <aj> i was just nice to some mormon doorknockers!!!
  24 # <Omnic> AJ?!?!
  25 # <aj> i know!!!!!
  26 # <Omnic> I'm gonna have to kick your ass when you come over
  27 # <Culus> aj: GET THE HELL OUT OF THE CABAL! :P
  28
  29 ################################################################################
  30
  31 import os, pg, string, sys;
  32 import utils;
  33 import apt_pkg;
  34
  35 ################################################################################
  36
  37 Cnf = None;
  38 projectB = None;
  39 ################################################################################
  40
  41 def usage (exit_code):
  42     print """Usage: julia [OPTION]... PASSWD_FILE
  43 Sync postgres' pg_shadow with PASSWD_FILE.
  44
  45   -v, --verbose explain what is being done
  46   -h, --help    display this help and exit"""
  47     sys.exit(exit_code)
  48
  49 ################################################################################
  50
  51 def main ():
  52     global Cnf, projectB;
  53
  54     apt_pkg.init();
  55
  56     Cnf = apt_pkg.newConfiguration();
  57     apt_pkg.ReadConfigFileISC(Cnf,utils.which_conf_file());
  58
  59     Arguments = [('q',"quiet","Julia::Options::Quiet"),
  60                  ('v',"verbose","Julia::Options::Verbose"),
  61                  ('D',"debug","Julia::Options::Debug", "IntVal"),
  62                  ('h',"help","Julia::Options::Help"),
  63                  ('V',"version","Julia::Options::Version")];
  64
  65     arguments = apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv);
  66     Options = Cnf.SubTree("Julia::Options")
  67
  68     if Options["Help"]:
  69         usage(0);
  70     if arguments == []:
  71         utils.warn("julia needs the name of the passwd file to sync with as an argument.");
  72         usage(1);
  73     elif len(arguments) > 1:
  74         utils.warn("julia only takes one non-option argument, the passwd file to sync with.");
  75         usage(1);
  76
  77     passwd = arguments[0];
  78     projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"], int(Cnf["DB::Port"]));
  79     valid_gid = Cnf.get("Julia::ValidGID","");
  80
  81     passwd_unames = {};
  82     passwd_file = utils.open_file(passwd);
  83     for line in passwd_file.readlines():
  84         split = string.split(line, ':');
  85         uname = split[0];
  86         gid = split[3];
  87         if valid_gid and gid != valid_gid:
  88             if Options["Verbose"]:
  89                 print "Skipping %s (GID %s != Valid GID %s)." % (uname, gid, valid_gid);
  90             continue;
  91         passwd_unames[uname] = "";
  92
  93     postgres_unames = {};
  94     q = projectB.query("SELECT usename FROM pg_user");
  95     ql = q.getresult();
  96     for i in ql:
  97         uname = i[0];
  98         postgres_unames[uname] = "";
  99
 100     known_postgres_unames = {};
 101     for i in string.split(Cnf.get("Julia::KnownPostgres",""),","):
 102         uname = string.strip(i);
 103         known_postgres_unames[uname] = "";
 104
 105     keys = postgres_unames.keys()
 106     keys.sort();
 107     for uname in keys:
 108         if not passwd_unames.has_key(uname)and not known_postgres_unames.has_key(uname):
 109             print "W: %s is in Postgres but not the passwd file or list of known Postgres users." % (uname);
 110
 111     keys = passwd_unames.keys()
 112     keys.sort();
 113     for uname in keys:
 114         if not postgres_unames.has_key(uname):
 115             if not Options["Quiet"]:
 116                 print "Creating %s user in Postgres." % (uname);
 117             q = projectB.query('CREATE USER "%s"' % (uname));
 118
 119 #######################################################################################
 120
 121 if __name__ == '__main__':
 122     main()
 123