test2backup.py

import sys
from sys import argv
from multiprocessing import Process
import os
script, directory = argv

#def convert_to_tmlearn(file, target):
#   line = target.read()
#   print "Read Line: %s" % (line)
#   return

dirs = os.listdir(directory)
def find_line(line, linelength):
    line
    linelength = linelength + len(line)
    line = target.readline(linelength)
    return (line, linelength)

def reorganize_tms(buffer2,buffer3,num_ones_double,past_tm10_flag):
# for some reason, this code won't work if used in a while true loop
    multiplier = 2
    if num_ones_double == 0:
        buffer3 = "0 "
        return (buffer3,multiplier)
    if past_tm10_flag == True:
        multiplier = 3

    buffer3 += buffer2[num_ones_double*multiplier - multiplier:]
    buffer2 = buffer2[:num_ones_double*multiplier - multiplier]
    num_ones_double -= 1
    buffer3 += buffer2[num_ones_double*multiplier - multiplier:]
    buffer2 = buffer2[:num_ones_double*multiplier - multiplier]
    num_ones_double -= 1
    buffer3 += buffer2[num_ones_double*multiplier - multiplier:]
    buffer2 = buffer2[:num_ones_double*multiplier - multiplier]
    num_ones_double -= 1
    buffer3 += buffer2[num_ones_double*multiplier - multiplier:]
    buffer2 = buffer2[:num_ones_double*multiplier - multiplier]
    num_ones_double -= 1
    buffer3 += buffer2[num_ones_double*multiplier - multiplier:]
    buffer2 = buffer2[:num_ones_double*multiplier - multiplier]
    num_ones_double -= 1
    buffer3 += buffer2[num_ones_double*multiplier - multiplier:]
    buffer2 = buffer2[:num_ones_double*multiplier - multiplier]
    num_ones_double -= 1
    buffer3 += buffer2[num_ones_double*multiplier - multiplier:]
    buffer2 = buffer2[:num_ones_double*multiplier - multiplier]
    num_ones_double -= 1
    buffer3 += buffer2[num_ones_double*multiplier - multiplier:]
    buffer2 = buffer2[:num_ones_double*multiplier - multiplier]
    num_ones_double -= 1
    return (buffer3,multiplier)

def convert_binary_to_tm_no(multiple_of_eight,line,past_tm10_flag,ifdef_flag):
    line_buffer = ""
    buffer2 = ""
    some_count = 0
    num_ones = 0
    tm09_flag = False
    past_tm10_flag = False
    for line in line:
        some_count += 1
        which_tm = (8 * (multiple_of_eight) + 9) - (some_count - 4)
        if line == "%":
            if ifdef_flag:
                line_buffer += "    "
                some_count -= 1
            line_buffer += "    tmlearn "
        elif line == "1":
            buffer2 += "%s," % (which_tm)
            if which_tm == 9:
                tm09_flag = True
            elif which_tm >= 10:
                past_tm10_flag = True
            num_ones += 1

    num_ones_double = num_ones
    buffer3 = ""
    (buffer3,multiplier) = reorganize_tms(buffer2,buffer3,num_ones_double,past_tm10_flag)
    tm09_thing = multiplier
    if tm09_flag == True:
        tm09_thing = 2
    line_buffer += buffer3[:tm09_thing+(num_ones-1)*multiplier-1] + "\n"
    ifdef_flag = False
    return (line_buffer,past_tm10_flag,ifdef_flag)

for file in dirs:
    if file == "test2backup.py":
        continue
    target = open(file, 'rw+')
    line = target.readline()
    linelength = 0
    line0 = line
    while True:
        (line, linelength) = find_line(line, linelength)
        if line == "; learnset\n":
            break

    (line, linelength) = find_line(line, linelength)
    saved_linelength = linelength
    main_line = ""
    multiple_of_eight = 0
    past_tm10_flag = False
    ifdef_flag = False
    multiple_copy = 0
    completed_ifdef = True
    completed_flag = False
    if_flag = False
    else_flag = False

    print file
    while not completed_flag:
        if line == "ENDC\n":
            #print "we did it"
            multiple_copy = 0
            main_line += "ENDC\n"
            completed_ifdef = True
            else_flag = False
            (line, linelength) = find_line(line, linelength)
            if line != "db 0 ; padding\n":
                (line, linelength) = find_line(line, linelength)
        if line == "IF DEF(_YELLOW)\n":
            main_line += "IF DEF(_YELLOW)\n"
            (line, linelength) = find_line(line, linelength)
            ifdef_flag = True
            if_flag = True
            completed_ifdef = False
            multiple_copy = multiple_of_eight
            anotherFlag = True
            continue
        elif line == "ELSE\n":
            multiple_of_eight = multiple_copy
            if_flag = False
            else_flag = True
            #print "it worked!"
            main_line += "ELSE\n"
            (line, linelength) = find_line(line, linelength)
            ifdef_flag = True
            continue
        elif if_flag:
            ifdef_flag = True
        elif else_flag:
            ifdef_flag = True
        if multiple_of_eight >= 7 and completed_ifdef == True:
            completed_flag = True
            #(line, linelength) = find_line(line, linelength)
            continue

        (line_buffer,past_tm10_flag,ifdef_flag) = convert_binary_to_tm_no(multiple_of_eight,line,past_tm10_flag,ifdef_flag)
        main_line += line_buffer
        multiple_of_eight += 1
        (line, linelength) = find_line(line, linelength)
        if multiple_of_eight >= 12:
            sys.exit("rekt")
    target.seek(linelength)
    line2 = line
    main_line += line2

    #print main_line
    target.seek(saved_linelength)
    target.write(main_line)
    target.close()