import os
import sys


# begin dealing the file
def seperate(filename, file_name_dealed):
    xxx = ''
    with open(filename) as cheng:
        for line in cheng.readlines()[:]:
            line = line.split("\t")
            start = int(line[1])
            end = int(line[2])
            line[2] = str(start + 50)
            for line_item in line[:5]:
                xxx += str(line_item) + "\t"
            xxx += line[5]
            line[1] = str(end - 50)
            line[2] = end
            if line[5] == "-\n":
                line[5] = "+"
            else:
                line[5] = "-"
            for line_item in line:
                xxx += str(line_item) + "\t"
            xxx += "\n"

    with open(file_name_dealed, 'a') as lyu:
        lyu.write(xxx)


# find all of files that we need to deal in the current directory
def cheng_search(path):
    cfile_deal = []
    cfilelists = os.listdir(str(path))
    for cfile in cfilelists:
        if os.path.splitext(cfile)[1] == ".bed":
            cfile_deal.append(cfile)
    return cfile_deal


def getfile_suffix(filename):
    filename_suffix = filename.split(".")
    return filename_suffix[0]


def main(path):
    cfile_deal_lists = cheng_search(path)
    for cfile_deal in cfile_deal_lists:
        seperate(cfile_deal, getfile_suffix(cfile_deal) + ".txt")


if __name__ == '__main__':
    main(sys.argv[1])