greenbone /
gvm-tools
| Conditions | 11 |
| Total Lines | 61 |
| Code Lines | 38 |
| Lines | 0 |
| Ratio | 0 % |
| Changes | 0 | ||
Small methods make your code easier to understand, in particular if combined with a good name. Besides, if your method is small, finding a good name is usually much easier.
For example, if you find yourself adding comments to a method's body, this is usually a good sign to extract the commented part to a new method, and use the comment as a starting point when coming up with a good name for this new method.
Commonly applied refactorings include:
If many parameters/temporary variables are present:
Complex classes like gvmtools.cli.main() often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.
Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.
| 1 | # -*- coding: utf-8 -*- |
||
| 47 | def main(): |
||
| 48 | parser = create_parser(description=HELP_TEXT, logfilename='gvm-cli.log') |
||
| 49 | |||
| 50 | parser.add_argument('-X', '--xml', help='XML request to send') |
||
| 51 | parser.add_argument('-r', '--raw', help='Return raw XML', |
||
| 52 | action='store_true', default=False) |
||
| 53 | parser.add_argument('infile', nargs='?', type=open, default=sys.stdin) |
||
| 54 | |||
| 55 | args = parser.parse_args() |
||
| 56 | |||
| 57 | # If timeout value is -1, then the socket has no timeout for this session |
||
| 58 | if args.timeout == -1: |
||
| 59 | args.timeout = None |
||
| 60 | |||
| 61 | xml = '' |
||
| 62 | |||
| 63 | if args.xml is not None: |
||
| 64 | xml = args.xml |
||
| 65 | else: |
||
| 66 | # If this returns False, then some data are in sys.stdin |
||
| 67 | if not args.infile.isatty(): |
||
| 68 | try: |
||
| 69 | xml = args.infile.read() |
||
| 70 | except (EOFError, BlockingIOError) as e: |
||
| 71 | print(e) |
||
| 72 | |||
| 73 | # If no command was given, program asks for one |
||
| 74 | if len(xml) == 0: |
||
| 75 | xml = input() |
||
| 76 | |||
| 77 | # Remove all newlines if the commands come from file |
||
| 78 | xml = xml.replace('\n', '').replace('\r', '') |
||
| 79 | |||
| 80 | # Ask for password if none are given |
||
| 81 | if args.gmp_username and not args.gmp_password: |
||
| 82 | args.gmp_password = getpass.getpass('Enter password for ' + |
||
| 83 | args.gmp_username + ': ') |
||
| 84 | |||
| 85 | connection = create_connection(**vars(args)) |
||
| 86 | |||
| 87 | if args.raw: |
||
| 88 | transform = None |
||
| 89 | else: |
||
| 90 | transform = CheckCommandTransform() |
||
| 91 | |||
| 92 | gvm = Gmp(connection, transform=transform) |
||
| 93 | |||
| 94 | if args.gmp_username: |
||
| 95 | gvm.authenticate(args.gmp_username, args.gmp_password) |
||
| 96 | |||
| 97 | try: |
||
| 98 | result = gvm.send_command(xml) |
||
| 99 | |||
| 100 | print(result) |
||
| 101 | except Exception as e: # pylint: disable=broad-except |
||
| 102 | print(e) |
||
| 103 | sys.exit(1) |
||
| 104 | |||
| 105 | gvm.disconnect() |
||
| 106 | |||
| 107 | sys.exit(0) |
||
| 108 | |||
| 112 |
