#!/usr/bin/env python 
# -*- coding: utf-8 -*-

## <config> ############################################
duplicateFilePath = u"/nas/020_VIDEO/duplicates-lin.txt"
locationFileName = u"ku-ku-lin.txt"
folderPathAll = u"/nas/020_VIDEO/090_All"
folderPathByFirstLetter = u"/nas/020_VIDEO/010_By_Letter"
folderPathByYear = u"/nas/020_VIDEO/050_By_Year"
sourceFolders = <\
    u"/nas/080_DRIVES/DriveD/VIDEO", \
    u"/nas/080_DRIVES/DriveE/VIDEO", \
    u"/nas/080_DRIVES/DriveF/VIDEO/00-Sorted", \
    u"/nas/080_DRIVES/DriveG/VIDEO", \
    u"/nas/080_DRIVES/DriveH/video", \
    u"/nas/080_DRIVES/DriveI/0001/Serials", \
    u"/nas/080_DRIVES/DriveI/0001/VIDEO", \
    u"/nas/080_DRIVES/DriveJ/VIDEO" >
## </config> ###########################################


import os,re

def os_symlink( a, b ):
    rp = os.path.relpath(a,os.path.dirname(b))
    os.symlink( rp , b )

def get_first_letter( fileName ):
    ltr = fileName<0>.upper()
    if "0123456789.,!@#$_()
{}-".count(ltr):
        return "(0-9)"
    return ltr

reYear = re.compile(r"(?<=\D)(?:19|20)\d{2}(?=\D)", re.UNICODE )

def get_years(fileName):
    return reYear.findall( fileName )
 

def clean_symlinks( folderPath ):
    for i in os.listdir( folderPath ):
        j = os.path.join( folderPath, i )
        if os.path.islink( j ):
            os.unlink( j )
        elif os.path.isdir( j ):
            clean_symlinks( j )
            if len( os.listdir( j ) ) == 0 :
                os.rmdir( j )
 
print 'Processing...'

clean_symlinks( folderPathAll )
clean_symlinks( folderPathByFirstLetter )
clean_symlinks( folderPathByYear )

knownFilms = {}
for f in sourceFolders:
    print f
    for i in os.listdir( f ):
        j = os.path.join( f, i )
        if os.path.isdir( j ):
            with open( os.path.join( j, locationFileName ) , 'w' ) as loc:
                loc.write( j.encode('utf-8') )
            key = i.upper( )
            if key in knownFilms :
                knownFilms<key><1>.append( j )
            else:
                knownFilms</key><key>=( i.strip() , < j > )

with open( duplicateFilePath , 'w' ) as loc:
    for i in knownFilms.values():
        if len(i<1>) != 1 :
            loc.write( ('### ' + i<0> + '\n\r').encode('utf-8') )
            for j in i<1> :
                loc.write( (j + '\n\r').encode('utf-8') )
            loc.write( '\n\r\n\r' )

for i in knownFilms.values():
    os_symlink( i<1><0> , os.path.join( folderPathAll , i<0> ) )
    f = os.path.join( folderPathByFirstLetter , get_first_letter( i<0> ) )
    if not os.path.exists( f ) :
        os.mkdir( f )
    os_symlink( i<1><0> , os.path.join( f , i<0> ) )
    dt = get_years( i<0> )
    for y in dt :
        f = os.path.join( folderPathByYear , y )
        if not os.path.exists( f ) :
            os.mkdir( f )
        os_symlink( i<1><0> , os.path.join( f , i<0> ) )

print 'Done!'</key>