Différences

Ci-dessous, les différences entre deux révisions de la page.

--- python:les_modules [2013/12/03 22:00] – [Chemin des modules] root
+++ python:les_modules [2024/12/20 12:41] (Version actuelle) – [Fabric] admin
@@ Ligne 198: / Ligne 198: @@
 #!/usr/bin/env python
-from fabric.api import *
+import sys, os, getpass
-from fabric.colors import *
+#import sys, os, getpass, multiprocessing
+try:
+        from fabric.api import *
+        from fabric.colors import *
+except:
+        print('Fabric module must be installed !')
+        sys.exit(1)
+sys.tracebacklimit = 0
+output['running'] = False
+output['stdout'] = False
+output['aborts'] = False
+output['status'] = False
+output['warnings'] = False
+#env.user = 'root'
+#env.password = None
-env.hosts = ['host1', 'toto@host2']
-env.user = 'root'
-env.password = 'xxx'
-#env.gateway = 'mygateway',
 #env.passwords = {
-    'user@host':'pass',
+#    'user@host':'pass',
-    'gwuser@gwhost':'gwpass'
+#    'gwuser@gwhost':'gwpass'
-}
+#}
 #env.shell = "/bin/sh -c"
+#env.gateway = None
+env.disable_known_hosts = True
+#env.parralel = True
+#env.pool_size = multiprocessing.cpu_count() #Sets the number of concurrent processes to use when executing tasks in parallel.
 #env.keepalive = 300
-#env.roledefs = {
+#env.timeout = 10 #Network ssh connection timeout, in seconds.
-    'web': ['www1', 'www2', 'www3'],
+#env.command_timeout = 60
-    'dns': ['ns1', 'ns2']
+env.linewise = True
+env.warn_only = True
+env.role_default = 'adm'
+env.roledefs = {
+    'eth': ['svr1', 'svr2', 'svr3', 'svr4', 'svr5', 'svr6', 'svr7', 'svr8', 'svr9', 'svr10', 'svr11', 'svr12', 'svr13', 'svr14', 'svr15', 'svr16'],
+    'adm': ['svr1-adm', 'svr2-adm', 'svr3-adm', 'svr4-adm', 'svr5-adm', 'svr6-adm', 'svr7-adm', 'svr8-adm', 'svr9-adm', 'svr10-adm', 'svr11-adm', 'svr12-adm', 'svr13-adm', 'svr14-adm', 'svr15-adm', 'svr16-adm'],
+    'ib': ['svr1-ib', 'svr2-ib', 'svr3-ib', 'svr4-ib', 'svr5-ib', 'svr6-ib', 'svr7-ib', 'svr8-ib', 'svr9-ib', 'svr10-ib', 'svr11-ib', 'svr12-ib', 'svr13-ib', 'svr14-ib', 'svr15-ib', 'svr16-ib'],
+    'imm': ['svr1-imm', 'svr2-imm', 'svr3-imm', 'svr4-imm', 'svr5-imm', 'svr6-imm', 'svr7-imm', 'svr8-imm', 'svr9-imm', 'svr10-imm', 'svr11-imm', 'svr12-imm', 'svr13-imm', 'svr14-imm', 'svr15-imm', 'svr16-imm'],
+    'web_adm': ['sw-eth', 'sw-ib'],
 }
-env.disable_known_hosts = True
-env.parralel = True
-env.pool_size = 0 #Sets the number of concurrent processes to use when executing tasks in parallel.
-env.timeout = 10 #Network ssh connection timeout, in seconds.
-env.command_timeout = 10
-#env.warn_only = True
-hide('warnings', 'running', 'stdout', 'stderr')
-def remote_info():
+env.roledefs['web_adm'] += env.roledefs['imm']
-        with hide('running', 'stdout'):
-                print(yellow("Executing on %(host)s as %(user)s" % env, bold=True))
-                a = run('uname -a')
-                fastprint(a, show_prefix=True)
-def local_info():
+if env.hosts == [] and env.roles == []:
-        local('uname -a')
+        env.hosts = env.roledefs[env.role_default]
+#list_all = []
+#for server in env.roledefs.values():
+#       list_all += server
+#env.roledefs['all'] = list(set(list_all))
+def VERIF_IPMI_PASSWORD():
+        if 'IPMI_PASSWORD' in os.environ:
+                globals()['IPMI_PASSWORD'] = os.environ['IPMI_PASSWORD']
+        if 'IPMI_PASSWORD' not in globals():
+                IPMI_PASSWORD = getpass.getpass(prompt='IMM password : ')
+                globals()['IPMI_PASSWORD'] = IPMI_PASSWORD
+@task
+@serial
+def lr():
+        """ : Liste les roles utilisables avec l'option '-R'"""
+        for role in env.roledefs:
+                if role == env.role_default:
+                        print("%s (default) : %s" % (role, ' ,'.join(env.roledefs[role])))
+                else:
+                        print("%s : %s" % (role, ' ,'.join(env.roledefs[role])))
+        sys.exit()
+#@parallel(pool_size = multiprocessing.cpu_count() + 1)
+@task
+def cmd(cmd, show = None):
+        """ : Run command on a host (show='details' for details, show='status' for see if command is successful or not)"""
+        #with show('stdout','running','warnings'):
+        try:
+                out = run(cmd)
+        except Exception as err:
+                sys.stdout.write(yellow("%(host)s (%(user)s) : " % env, bold = True))
+                print(red(err))
+                return
+        if show == None:
+                for line in out.splitlines():
+                        sys.stdout.write(yellow("%(host)s (%(user)s) : " % env, bold = True))
+                        print(line)
+                if out.failed:
+                        sys.stdout.write(yellow("%(host)s (%(user)s) : " % env, bold = True))
+                        print(red(' KO', bold = True)) + " (return %s)" % out.return_code
+        elif show == 'details' or show == 'status':
+                sys.stdout.write(yellow("Executing on %(host)s as %(user)s : " % env, bold = True) + blue(cmd, bold = True))
+                if out.failed:
+                        print(red(' KO', bold = True))
+                else:
+                        print(green(' OK', bold = True))
+                if show ==  'details':
+                        print(yellow('OUT : (return code = %s)' % out.return_code))
+                        print('%s\n' % out)
+        else:
+                print("Option %s unknow" % show)
+                sys.exit(1)
+@task
+def send_file(src, dst = None, mode = None):
+        """ : Send file on a host (dst=<destination>, mode=<mode>)"""
+        if dst == None:
+                dst = src
+        if mode == None:
+                put(src, dst)
+        else:
+                try:
+                        int(mode)
+                except:
+                        print("'mode' value must be an integer !")
+                        sys.exit(1)
+                put(src, dst, mode=mode)
+@task
+def reboot(wait = 0):
+        """ : Reboot a host (wait=<secondes>)"""
+        if os.getlogin() != 'root':
+                print('Only root must be reboot a host')
+                sys.exit(1)
+        try:
+                int(wait)
+        except:
+                print("'wait' value must be an integer !")
+                sys.exit(1)
+        reboot(wait = wait)
+@serial
+@task
+def power(state):
+        """ : IMM power manager (state=<status|on|off|cycle|reset|diag|soft>)"""
+        if not env.host in env.roledefs['imm']:
+                print('Use "-R imm" or "-H <host>-imm"')
+                print('%s not in this table : %s' % (env.host, ','.join(env.roledefs['imm'])))
+                sys.exit(1)
+        if state not in ['status','on','off','cycle','reset','diag','soft']:
+                print("%s state is not supported. Please use state <status|on|off|cycle|reset|diag|soft> !" % state)
+                sys.exit(1)
+        VERIF_IPMI_PASSWORD()
+        out = local("ipmitool -I lanplus -C 2 -U system -P '%s' -E -H %s chassis power %s" % (globals()['IPMI_PASSWORD'], env.host, state), capture=True)
+        if out.failed:
+                sys.stdout.write(yellow("%(host)s : " % env, bold = True))
+                print(red('Failed ! (return code : %s)' % out.return_code))
+                print(out.stderr)
+                sys.exit(1)
+        else:
+                for line in out.splitlines():
+                        sys.stdout.write(yellow("%(host)s : " % env, bold = True))
+                        print(line)
+@serial
+@task
+def console(action='activate'):
+        """ : IMM SOL (action=<activate|deactivate>)"""
+        if len(env.hosts) != 1:
+                print('Only on one host. Use option -H to precise the host !')
+                sys.exit(1)
+        if not env.host in env.roledefs['imm']:
+                print('Use "-H <host>-imm"')
+                print('%s not in this table : %s' % (env.host, ','.join(env.roledefs['imm'])))
+                sys.exit(1)
+        VERIF_IPMI_PASSWORD()
+        os.system("ipmitool -I lanplus -C 2 -U system -P '%s' -E -H %s sol %s" % (globals()['IPMI_PASSWORD'], env.host, action))
+@serial
+@task
+def web_adm():
+        """ : Open Firefox to administrate (use -H option)"""
+        if len(env.hosts) != 1:
+                print('Only on one host. Use option -H to precise the host (%s)!' % ', '.join(env.roledefs['web_adm']))
+                sys.exit(1)
+        if os.getenv('DISPLAY') == None:
+                print('You must have a valid display !')
+                sys.exit(1)
+        if env.host not in env.roledefs['web_adm']:
+                print("%s is not a valid host ! Valid host are %s" % (env.host, ', '.join(env.roledefs['web_adm'])))
+                sys.exit(1)
+        os.system('firefox https://%s/' % env.host)
 </code>
+===== readline =====
+<code python readline.py>
+import readline, signal
+prompt = 'GigiX > '
+def signal_handler(signal, frame):
+    print
+    input(prompt)
+    return
+signal.signal(signal.SIGINT, signal_handler)
+readline.parse_and_bind("tab: complete")
+class completer:
+        def __init__(self, tree):
+            self.tree = tree
+        def _parser(self, line, search):
+            if (search is None) or (callable(search) == True):
+                return []
+            elif len(line) == 0:
+                return [x for x in search]
+            elif len(line) == 1 and readline.get_line_buffer()[-1] != ' ':
+                return ['%s ' % x for x in search if x.startswith(line[0])]
+            else:
+                return self._parser(line[1:], search[line[0]])
+            return []
+        def complete(self, text, state):
+            line = readline.get_line_buffer().split()
+            results = self._parser(line, self.tree)
+            return results[state]
+def linux_sles_12():
+    print('Je suis Linux SLES 12')
+tree = {
+    'unix': {
+        'solaris': {
+            '8':None,
+            '9':None,
+            '10':None,
+            '11':None,
+        }
+    },
+    'linux': {
+        'debian': {
+            'buzz':None,
+            'rex':None,
+            'bo':None,
+            'hamm':None,
+            'slink':None,
+            'patato':None,
+            'woody':None,
+            'sarge':None,
+            'etch':None,
+            'lenny':None,
+            'squeeze':None,
+            'wheezy':None,
+            'jessie':None,
+    },
+        'RHEL': {
+            '3':None,
+            '4':None,
+            '5':None,
+            '6':None,
+            '7':None,
+        },
+        'SLES': {
+            '9':None,
+            '10':None,
+            '11':None,
+            '12':linux_sles_12,
+        }
+    },
+    'windows': {
+        'xp':None,
+        'vista':None,
+        'seven':None,
+        '8':None,
+        '10':None,
+    }
+}
+readline.set_completer(completer(tree).complete)
+l = ''
+while l != ('quit' or 'exit'):
+    try:
+            l = input(prompt)
+            eval("tree['"+"']['".join(l.split())+"']()")
+    except EOFError:
+            print('Bye')
+            break
+    except:
+        continue
+</code>
 ===== Twisted =====
@@ Ligne 259: / Ligne 527: @@
 import ctypes, ctypes.util
-ctypes.cdll.LoadLibrary(ctypes.util.find_library('c'))
+libc=ctypes.cdll.LoadLibrary(ctypes.util.find_library('c'))
 libc.printf("uid = %d - gid = %d\n", libc.geteuid(), libc.getgid())
 </code>