0lever-so 1.9.1__tar.gz

0lever_so-1.9.1/0lever_so.egg-info/PKG-INFO

@@ -0,0 +1,117 @@
+Metadata-Version: 2.4
+Name: 0lever-so
+Version: 1.9.1
+Summary: 服务器登录工具
+Home-page: https://github.com/0lever/so
+Author: fqiyou
+Author-email: yc.fqiyou@gmail.com
+Requires-Dist: pyyaml
+Requires-Dist: pexpect
+Dynamic: author
+Dynamic: author-email
+Dynamic: description
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: summary
+
+======
+so
+======
+
+This is a SSH login tool
+
+Installation
+============
+
+::
+
+    pip install --upgrade 0lever-so
+    or
+    pip install --upgrade 0lever-so -i https://pypi.org/simple/
+
+
+Usage
+=====
+
+::
+
+    # 初始化配置文件,升级无需初始化,chmod 400 ~/.so/keys/*
+    ➜  ~ so_install
+    ➜  ~ cd .so
+    ➜  .so tree
+    .
+    ├── keys
+    │   └── demo.pem
+    └── password.yaml
+
+    1 directory, 2 files
+    ➜  .so
+
+
+::
+
+    # 配置文件
+    ssh:
+      - id: 1
+        name: demo1
+        user: fqiyou
+        password: xxx
+        host: 1.1.1.1
+        port: 20755
+      - id: 2
+        name: demo2
+        user: fqiyou
+        password: xxx
+        host: 1.1.1.1
+        port: 39986
+      - id: 3
+        name: demo3
+        user: root
+        password: demo.pem
+        host: 1.1.1.1
+        port: 22
+
+
+Other-shell
+
+=====
+
+::
+
+    #!/usr/bin/expect
+    set USER "xxx"
+    set PASSWD "xxx"
+    set timeout 10
+
+    trap {
+        set rows [stty rows]
+        set cols [stty columns]
+        stty rows $rows columns $cols < $spawn_out(slave,name)
+    } WINCH
+    spawn su - $USER
+    expect "Password: "
+    send "$PASSWD\n"
+    interact
+
+::
+
+    #!/usr/bin/expect -f
+    set HOST [lindex $argv 0]
+    set USER [lindex $argv 1]
+    set PASSWD [lindex $argv 2]
+    set PORT [lindex $argv 3]
+    set timeout 10
+
+    trap {
+        set rows [stty rows]
+        set cols [stty columns]
+        stty rows $rows columns $cols < $spawn_out(slave,name)
+    } WINCH
+
+    spawn ssh $USER@HOST -p $PORT
+    expect {
+        "*yes/no" {send "yes\r"; exp_continue}
+        "*password:" {send "$PASSWD\r"}
+    }
+    interact
+    ```

0lever_so-1.9.1/0lever_so.egg-info/SOURCES.txt

@@ -0,0 +1,10 @@
+README.rst
+setup.py
+0lever_so.egg-info/PKG-INFO
+0lever_so.egg-info/SOURCES.txt
+0lever_so.egg-info/dependency_links.txt
+0lever_so.egg-info/entry_points.txt
+0lever_so.egg-info/requires.txt
+0lever_so.egg-info/top_level.txt
+so/__init__.py
+so/so.py

0lever_so-1.9.1/0lever_so.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

0lever_so-1.9.1/0lever_so.egg-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+so = so.so:run
+so_install = so.so:run_install

0lever_so-1.9.1/0lever_so.egg-info/requires.txt

@@ -0,0 +1,2 @@
+pyyaml
+pexpect

0lever_so-1.9.1/0lever_so.egg-info/top_level.txt

@@ -0,0 +1 @@
+so

0lever_so-1.9.1/PKG-INFO

@@ -0,0 +1,117 @@
+Metadata-Version: 2.4
+Name: 0lever-so
+Version: 1.9.1
+Summary: 服务器登录工具
+Home-page: https://github.com/0lever/so
+Author: fqiyou
+Author-email: yc.fqiyou@gmail.com
+Requires-Dist: pyyaml
+Requires-Dist: pexpect
+Dynamic: author
+Dynamic: author-email
+Dynamic: description
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: summary
+
+======
+so
+======
+
+This is a SSH login tool
+
+Installation
+============
+
+::
+
+    pip install --upgrade 0lever-so
+    or
+    pip install --upgrade 0lever-so -i https://pypi.org/simple/
+
+
+Usage
+=====
+
+::
+
+    # 初始化配置文件,升级无需初始化,chmod 400 ~/.so/keys/*
+    ➜  ~ so_install
+    ➜  ~ cd .so
+    ➜  .so tree
+    .
+    ├── keys
+    │   └── demo.pem
+    └── password.yaml
+
+    1 directory, 2 files
+    ➜  .so
+
+
+::
+
+    # 配置文件
+    ssh:
+      - id: 1
+        name: demo1
+        user: fqiyou
+        password: xxx
+        host: 1.1.1.1
+        port: 20755
+      - id: 2
+        name: demo2
+        user: fqiyou
+        password: xxx
+        host: 1.1.1.1
+        port: 39986
+      - id: 3
+        name: demo3
+        user: root
+        password: demo.pem
+        host: 1.1.1.1
+        port: 22
+
+
+Other-shell
+
+=====
+
+::
+
+    #!/usr/bin/expect
+    set USER "xxx"
+    set PASSWD "xxx"
+    set timeout 10
+
+    trap {
+        set rows [stty rows]
+        set cols [stty columns]
+        stty rows $rows columns $cols < $spawn_out(slave,name)
+    } WINCH
+    spawn su - $USER
+    expect "Password: "
+    send "$PASSWD\n"
+    interact
+
+::
+
+    #!/usr/bin/expect -f
+    set HOST [lindex $argv 0]
+    set USER [lindex $argv 1]
+    set PASSWD [lindex $argv 2]
+    set PORT [lindex $argv 3]
+    set timeout 10
+
+    trap {
+        set rows [stty rows]
+        set cols [stty columns]
+        stty rows $rows columns $cols < $spawn_out(slave,name)
+    } WINCH
+
+    spawn ssh $USER@HOST -p $PORT
+    expect {
+        "*yes/no" {send "yes\r"; exp_continue}
+        "*password:" {send "$PASSWD\r"}
+    }
+    interact
+    ```

0lever_so-1.9.1/README.rst

@@ -0,0 +1,101 @@
+======
+so
+======
+
+This is a SSH login tool
+
+Installation
+============
+
+::
+
+    pip install --upgrade 0lever-so
+    or
+    pip install --upgrade 0lever-so -i https://pypi.org/simple/
+
+
+Usage
+=====
+
+::
+
+    # 初始化配置文件,升级无需初始化,chmod 400 ~/.so/keys/*
+    ➜  ~ so_install
+    ➜  ~ cd .so
+    ➜  .so tree
+    .
+    ├── keys
+    │   └── demo.pem
+    └── password.yaml
+
+    1 directory, 2 files
+    ➜  .so
+
+
+::
+
+    # 配置文件
+    ssh:
+      - id: 1
+        name: demo1
+        user: fqiyou
+        password: xxx
+        host: 1.1.1.1
+        port: 20755
+      - id: 2
+        name: demo2
+        user: fqiyou
+        password: xxx
+        host: 1.1.1.1
+        port: 39986
+      - id: 3
+        name: demo3
+        user: root
+        password: demo.pem
+        host: 1.1.1.1
+        port: 22
+
+
+Other-shell
+
+=====
+
+::
+
+    #!/usr/bin/expect
+    set USER "xxx"
+    set PASSWD "xxx"
+    set timeout 10
+
+    trap {
+        set rows [stty rows]
+        set cols [stty columns]
+        stty rows $rows columns $cols < $spawn_out(slave,name)
+    } WINCH
+    spawn su - $USER
+    expect "Password: "
+    send "$PASSWD\n"
+    interact
+
+::
+
+    #!/usr/bin/expect -f
+    set HOST [lindex $argv 0]
+    set USER [lindex $argv 1]
+    set PASSWD [lindex $argv 2]
+    set PORT [lindex $argv 3]
+    set timeout 10
+
+    trap {
+        set rows [stty rows]
+        set cols [stty columns]
+        stty rows $rows columns $cols < $spawn_out(slave,name)
+    } WINCH
+
+    spawn ssh $USER@HOST -p $PORT
+    expect {
+        "*yes/no" {send "yes\r"; exp_continue}
+        "*password:" {send "$PASSWD\r"}
+    }
+    interact
+    ```

0lever_so-1.9.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

0lever_so-1.9.1/setup.py

@@ -0,0 +1,41 @@
+# -*- coding: utf-8 -*-
+
+from setuptools import setup, find_packages
+
+setup(name='0lever-so',
+
+      version="1.9.1",
+
+      url='https://github.com/0lever/so',
+
+      author='fqiyou',
+
+      author_email='yc.fqiyou@gmail.com',
+
+      description=u'服务器登录工具',
+
+      install_requires=["pyyaml", "pexpect"],
+
+      packages=find_packages(),
+
+      long_description=open('README.rst').read(),
+
+      package_data={
+      },
+
+      entry_points={
+            'console_scripts': [
+                  'so = so.so:run',
+                  'so_install = so.so:run_install',
+            ],
+      }
+
+)
+
+# pip install build twine
+# python -m build            # 生成 dist/ 下的 sdist 和 wheel
+# twine upload dist/* -r coohua
+# twine upload dist/* -r pypi
+# pip install --upgrade 0lever-so -i https://pypi.org/simple/
+
+# python -m build && twine upload dist/* -r pypi

0lever_so-1.9.1/so/so.py

@@ -0,0 +1,195 @@
+# -*- coding: utf-8 -*-
+import sys
+import pexpect
+import signal
+import yaml
+import os
+import shutil
+import time
+import termios
+import struct
+import fcntl
+
+password_yaml = """ssh: 
+  - id: 1
+    name: demo1
+    user: fqiyou
+    password: xxx
+    host: 1.1.1.1
+    port: 20755
+  - id: 2
+    name: demo2
+    user: fqiyou
+    password: xxx
+    host: 1.1.1.1
+    port: 39986
+  - id: 3
+    name: demo3
+    user: root
+    password: demo.pem
+    host: 1.1.1.1
+    port: 22
+"""
+keys_demo_pem = """-----BEGIN RSA PRIVATE KEY-----
+xxxxx
+-----END RSA PRIVATE KEY-----
+"""
+
+child = None
+termios_size = None
+def _sigwinch_passthrough(sig, data):
+    global child,termios_size
+    if 'TIOCGWINSZ' in dir(termios):
+        TIOCGWINSZ = termios.TIOCGWINSZ
+    else:
+        TIOCGWINSZ = 1074295912
+    s = struct.pack("HHHH", 0, 0, 0, 0)
+    a = struct.unpack('HHHH', fcntl.ioctl(sys.stdout.fileno(), TIOCGWINSZ, s))
+    termios_size = a
+    if child is not None:
+        child.setwinsize(a[0], a[1])
+
+
+def _exit(*args, **kwargs):
+    print('You choose to stop so.')
+    if child is not None:
+        child.close()
+    sys.exit(1)
+
+
+def _get_hosts_by_config(name=''):
+    try:
+        so_dir = os.path.join(os.path.expanduser('~'), ".so")
+        password_yaml_file = os.path.join(so_dir, "password%s.yaml" % ('-' + name if name else ''))
+        keys_dir = os.path.join(so_dir, "keys")
+        return yaml.load(open(password_yaml_file, 'r').read(), Loader=yaml.Loader)["ssh"], keys_dir
+    except Exception as e:
+        print(u"获取配置文件错误", e)
+        exit(1)
+        return
+
+def _login_ssh(user, password, host, port, oauth=False):
+    global child
+    if password.endswith('.pem'):
+        child = pexpect.spawn('ssh %s@%s -p %s -i %s' % (user, host, port, password))
+    elif password == '':
+        # 免密登陆，仅使用公钥认证
+        child = pexpect.spawn('ssh %s@%s -p %s -o BatchMode=yes -o PreferredAuthentications=publickey' % (user, host, port))
+    else:
+        child = pexpect.spawn('ssh %s@%s -p %s' % (user, host, port))
+    patterns = ['nodename nor servname provided', 'Connection refused'
+                , pexpect.TIMEOUT
+                , '[Pp]assword:'
+                , 'continue connecting (yes/no)?'
+                , '#', '~', 'Welcome']
+    if oauth:
+        patterns.append('请输入 OAuth 二次验证码')
+    i = child.expect(patterns)
+    if i <= 2:
+        print(child.before, child.after)
+        return
+    elif i == 3:
+        child.sendline(password)
+    elif i == 4:
+        child.sendline('yes')
+        if oauth:
+            import pyotp
+            key = str(pyotp.TOTP(password).now())
+            child.sendline(key)
+        else:
+            child.expect('[Pp]assword:')
+            child.sendline(password)
+    elif i in (5, 6, 7):
+        pass
+    elif oauth and i == 8:
+        import pyotp
+        key = str(pyotp.TOTP(password).now())
+        child.sendline(key)
+    else:
+        print(i)
+        print(child.before, child.after)
+        return
+    print('Login Success!')
+    child.sendline('')
+    _sigwinch_passthrough(None, None)
+    if termios_size is not None:
+        child.setwinsize(termios_size[0], termios_size[1])
+    child.interact()
+
+def _print_head():
+    print("##############################################")
+    print("\033[0;36m               SU Login Platform              \033[0m")
+    print("##############################################")
+
+def _print_underline():
+    print("----------------------------------------------")
+
+def _print_remind():
+    print("[*] 选择主机:")
+
+def _get_cmd_args():
+    remind = "[*] 选择主机:"
+    args = ""
+    if 2 == sys.version_info[0]:
+        args = raw_input(remind)
+    elif 3 == sys.version_info[0]:
+        args = input(remind)
+    else:
+        _print_remind()
+        args = sys.stdin.readline()
+    return str(args).replace("\n", "")
+
+
+def _print_host_list(host_list):
+    _print_underline()
+    print("  序号  |       主机      |   说明   " )
+    _print_underline()
+    for i in host_list:
+        print("\033[0;31m%4s\033[0m\t| %15s | %s\t" % (i['id'], i['host'], i['name']))
+    _print_underline()
+
+def _login(info, keys_dir):
+    user = info["user"]
+    password = info["password"]
+    host = info["host"]
+    port = info["port"]
+    if "key_type" in info.keys() and info["key_type"] == "OAuth":
+        _login_ssh(user=user, password=password, host=host, port=port, oauth=True)
+    else:
+        if password.endswith('.pem'):
+            password = os.path.join(keys_dir, password)
+        _login_ssh(user=user, password=password, host=host, port=port)
+
+def run():
+    name = sys.argv[1] if len(sys.argv) > 1 else ''
+    config_host_list, key_dir = _get_hosts_by_config(name)
+    config_host_map = dict(zip([str(i["id"]) for i in config_host_list], config_host_list))
+    # 信号
+    signal.signal(signal.SIGINT, _exit)
+    signal.signal(signal.SIGTERM, _exit)
+    signal.signal(signal.SIGWINCH, _sigwinch_passthrough)
+    _print_head()
+    while True:
+        _print_host_list(config_host_list)
+        # _print_remind()
+        # cmd_args = sys.stdin.readline().replace("\n", "")
+        cmd_args = _get_cmd_args()
+        if cmd_args in config_host_map:
+            _login(config_host_map[cmd_args], key_dir)
+        elif cmd_args == "q" or cmd_args == "exit" or cmd_args == "quit":
+            _exit()
+            return
+        else:
+            print("未知参数:%s" % cmd_args)
+
+
+def run_install():
+    so_dir = os.path.join(os.path.expanduser('~'), ".so")
+    if os.path.exists(so_dir):
+        shutil.move(so_dir, so_dir+"_" + str(int(time.time() * 1000)))
+    os.mkdir(so_dir)
+    os.mkdir(os.path.join(so_dir, "keys"))
+    with open(os.path.join(so_dir, 'password.yaml'), 'w') as f:
+        f.write(password_yaml)
+    with open(os.path.join(so_dir, "keys", 'demo.pem'), 'w') as f:
+        f.write(password_yaml)