vdb.py 文件源码

def main():
  import argparse
  parser = argparse.ArgumentParser()
  parser.add_argument('--pid', '-p', type=int, default=None)
  db = parser.add_mutually_exclusive_group(required=False)
  db.add_argument('--rpdb2', '-r', default=False, action='store_true',
                  help='Attach using rpdb2')
  db.add_argument('--rpdb', default=False, action='store_true',
                  help='Attach using rpdb (Client not implemented, use putty)')
  db.add_argument('--winpdb', '--gui', '-g', default=False, action='store_true',
                  help='Attach using winpdb')
  parser.add_argument('--ip', default='127.0.0.1',
                      help='Set ip address for rpdb/rpdb2/winpdb to attach on')
  parser.add_argument('--port', default=4444, type=int,
                      help='Set port for rpdb to attach on')
  parser.add_argument('--password', '--pw', default='vsi')
  parser.add_argument('args', nargs='*', 
                      help='Command to run with vdb attached. Not implemented yet')
  args = parser.parse_args()

  if args.pid:
    #attach to a pid
    if args.rpdb2 or args.winpdb:
      from .vdb_rpdb2 import attach as rpdb2_attach
      rpdb2_attach(args.pid, password=args.password, ip=args.ip, gui=args.winpdb)
    elif args.rpdb:
      from .vdb_rpdb import attach as rpdb_attach
      rpdb_attach(args.pid, ip=args.ip, port=args.port)
    else:
      attach(args.pid)
  else:
    pass #Do whatever pdb does to run the command
    #Copy pdb.main or ipdb.main