CrackNinja/FlagsNArgs.py at main · Adi2K/CrackNinja · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Wed Dec  2 12:40:00 2020

@author: aditya
"""

from Tokenize import Tokenize_space

def look_for_flags(inp):
    flags = ""
    tokens = Tokenize_space(inp)
    for x in tokens:
        if (x[0] == '-'):
            for i in range (1,len(x)):
                flags = flags+x[i]
    return flags


def look_for_arguments(inp):
    arguments = ""
    tokens = Tokenize_space(inp)
    for x in range(1,len(tokens)):
        if (tokens[x][0] != '-'):
            arguments = arguments+tokens[x]+" "
    return arguments


def look_for_main_arg(inp):
    mainarg = ""
    tokens = Tokenize_space(inp)
    i = 1
    token = tokens[i]
    while (token[0] != '-'):
        mainarg  =mainarg+token+" "
        i = i+1
        token = tokens[i]
    mainarg = mainarg.strip()
    return mainarg


def look_for_flag_arg_pair(inp):
    flag_arg_pair = {}
    tokens = Tokenize_space(inp)
    for token in tokens:
        if(token[0] == '-'):
            flag_arg_pair[token] = ""

    for x in range(1,len(tokens)):
        if tokens[x][0] == "-":
            i=1
            finalarg = ""
            if(x+i<len(tokens)):
                temparg = tokens[x+i]
            else:
                break
            while(temparg[0] != '-'):
                finalarg = finalarg + temparg + " "
                i = i+1
                if(x+i<len(tokens)):
                    temparg = tokens[x+i]
                else:
                    break
            flag_arg_pair[tokens[x]] = finalarg.strip()

    return flag_arg_pair


# try1 = "wishlist diwali wish -s -a hello world -r valorant part 2"
# ans = look_for_flag_arg_pair(try1)
# print(ans)