stitchvariable/stitchvariable.py at master · quickutils/stitchvariable · 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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
import sys
import argparse
import os
import datetime
from key_value_db import KeyValueDB, KeyValueObject

backup_path =  os.getenv('TMP') + "/stitchvariable/" if not os.getenv('TMP') == "" else "../tmp/stitchvariable/"
if not os.path.exists(backup_path):
    os.makedirs(backup_path)

class Ops:
    Unknown    = 0
    ParsingVar = 1

def change_variable_name(source_file_path, type, new_name):
    content = file_reader(source_file_path) + " "

    variables_to_modify = KeyValueDB()
    operation = Ops.Unknown
    line_ended = False
    last_token = ""
    previous_identifier = ""
    current_identifier = ""
    punctuations = "`¬!\"£$%^&*()_-=+\\|,<.>/?;:'@#~]}[{"
    new_content = ""
    for index in range(0, len(content) - 1):
        char = content[index]
        while index < len(content) - 1 and content[index].isspace():
            index += 1
            continue
        next_char = (content[index+1] if (index < len(content) - 1) else char)
        if char.isalnum():
            current_identifier += char
        elif char == ";" or char == "=":
            new_variable_name = current_identifier
            if (operation == Ops.ParsingVar or previous_identifier == type) and not current_identifier == "":
                new_variable_name = wild_card_resolver(new_name, current_identifier)
                variables_to_modify.add(current_identifier, new_variable_name)
            line_ended = True
            new_content += new_variable_name + char
            current_identifier = ""
            previous_identifier = ""
            operation = Ops.Unknown
        else:
            if current_identifier != "":
                new_variable_name = current_identifier
                if previous_identifier == type and last_token not in punctuations:
                    operation = Ops.ParsingVar
                    new_variable_name = wild_card_resolver(new_name, current_identifier)
                    variables_to_modify.add(current_identifier, new_variable_name)
                elif operation == Ops.ParsingVar:
                    new_variable_name = wild_card_resolver(new_name, current_identifier)
                    variables_to_modify.add(current_identifier, new_variable_name)
                new_variable_name = variables_to_modify.get(current_identifier)
                if not new_variable_name == "" and char not in "(":
                    new_content += new_variable_name
                else:
                    new_content += current_identifier
                previous_identifier = current_identifier
            current_identifier = ""
            last_token = char
            new_content += char

    for variable in variables_to_modify:
        print(variable.key, "=>", variable.value)
    file_writer(source_file_path, new_content)
    file_backup_path = file_backup(os.path.basename(source_file_path), content)
    print()
    print("Modified the variable names in", os.path.basename(source_file_path))
    print("Backup created at", file_backup_path)

def wild_card_resolver(raw_str, master_str):
    resolved_str = ""
    if len(raw_str) > 0:
        index = 0
        if raw_str[0] == '%':
            resolved_str = master_str + raw_str[1:len(raw_str)]
            index += 1
        if raw_str[len(raw_str)-1] == '%':
            resolved_str = resolved_str[0:len(resolved_str)-1] + master_str
    return resolved_str

def file_reader(source_file_path):
    if not os.path.isfile(source_file_path):
        print(source_file_path, "does not exist")
        exit(1)
    return open(source_file_path, "r").read()

def file_writer(source_file_path, content):
    f = open(source_file_path, "w+")
    f.write(content)
    f.close()

def file_backup(file_name, content):
    file_backup_path = os.path.realpath(backup_path) + os.sep + file_name + datetime.datetime.today().strftime('%m-%d-%Y-%H-%M-%S') + ".java"
    file_writer(file_backup_path, content)
    return file_backup_path

def screw_the_offside_rule():
    print(end='')

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Initiate a variable name change on a source file, support wildcard in variable name manipulation.')
    parser.add_argument('source_file_path', help='The value to convert')
    parser.add_argument('type', help='The value to convert')
    parser.add_argument('new_name', help='The value to convert')

    args = parser.parse_args()
    change_variable_name(args.source_file_path, args.type, args.new_name)