Log fragmenting tool

log_fragmenter.py

# Log fragmenting 'tool'
# Author: Julian Capilla-krumbak
# Date: 2016-05-04
#
# Expected log format: yyyymmdd_xxxxxxxxxxxx
# It will put logs in folders per day like ./20160101/ ./20160102/ .. etc

import os,shutil

class FolderDist(object):
    def __init__(self):
        files = input("Enter path to files[{}]: ".format(os.getcwd()))
        if files:
            path = files
        else:
            path = os.getcwd()
        months = ['01','02','03','04','05','06','07','08','09','10','11','12']
        for i in months:
            wpath = os.path.join(path,i)
            if os.path.exists(wpath):
                for x in os.listdir(wpath):
                    file = os.path.join(wpath,x)
                    newDir = os.path.join(wpath, x.split('_')[0])
                    if not os.path.exists(newDir):
                        os.mkdir(newDir)
                    if file != newDir:
                        print("Moving {} -> {}".format(file,newDir))
                        shutil.move(file,newDir)

if __name__ == '__main__':
    test = FolderDist()