sintonia/utils/airtime-import/airtime-import.py

#!/usr/local/bin/python
import sys
import os
import logging
from configobj import ConfigObj
import argparse
from argparse import RawTextHelpFormatter
from api_clients import api_client
import json
import shutil

# create logger
logger = logging.getLogger()

# no logging
ch = logging.NullHandler()

# add ch to logger
logger.addHandler(ch)


# loading config file
try:
    config = ConfigObj('/etc/airtime/media-monitor.cfg')
except Exception, e:
    print('Error loading config file: %s', e)
    sys.exit()

api_client = api_client.api_client_factory(config)

# action call back classes
class CopyAction(argparse.Action):
    def __call__(self, parser, namespace, values, option_string=None):
        stor = helper_get_stor_dir()
        if(stor is None):
            exit("Unable to connect to the server.")
        dest = stor+"organize/"
        copy_or_move_files_to(values, dest, 'copy')

class MoveAction(argparse.Action):
    def __call__(self, parser, namespace, values, option_string=None):
        stor = helper_get_stor_dir()
        if(stor is None):
            exit("Unable to connect to the server.")
        dest = stor+"organize/"
        copy_or_move_files_to(args.path, dest, 'move')

class WatchAddAction(argparse.Action):
    def __call__(self, parser, namespace, values, option_string=None):
        if(os.path.isdir(values)):
            res = api_client.add_watched_dir(values)
            if(res is None):
                exit("Unable to connect to the server.")
            # sucess
            if(res['msg']['code'] == 0):
                print "%s added to watched folder list successfully" % values
            else:
                print "Adding a watched folder failed. : %s" % res['msg']['error']
        else:
            print "Given path is not a directory: %s" % values
        
class WatchListAction(argparse.Action):
    def __call__(self, parser, namespace, values, option_string=None):
        res = api_client.list_all_watched_dirs()
        if(res is None):
            exit("Unable to connect to the server.")
        dirs = res["dirs"].items()
        # there will be always 1 which is storage folder
        if(len(dirs) == 1):
                print "No watch folders found"
        else:
            for key, value in dirs:
                if(key != '1'):
                    print value

class WatchRemoveAction(argparse.Action):
    def __call__(self, parser, namespace, values, option_string=None):
        if(os.path.isdir(values)):
            res = api_client.remove_watched_dir(values)
            if(res is None):
                exit("Unable to connect to the server.")
            # sucess
            if(res['msg']['code'] == 0):
                print "%s removed from watched folder list successfully" % values
            else:
                print "Removing a watched folder failed. : %s" % res['msg']['error']
        else:
            print "Given path is not a directory: %s" % values
            
class StorageSetAction(argparse.Action):
    def __call__(self, parser, namespace, values, option_string=None):
        if(os.path.isdir(values)):
            res = api_client.set_storage_dir(values)
            if(res is None):
                exit("Unable to connect to the server.")
            # sucess
            if(res['msg']['code'] == 0):
                print "Successfully set storage folder to %s" % values
            else:
                print "Setting storage folder to failed.: %s" % res['msg']['error']
        else:
            print "Given path is not a directory: %s" % values

class StorageGetAction(argparse.Action):
    def __call__(self, parser, namespace, values, option_string=None):
        print helper_get_stor_dir()

#helper functions
# copy or move files
# flag should be 'copy' or 'move'
def copy_or_move_files_to(paths, dest, flag):
    for path in paths:
        if(os.path.exists(path)):
            if(os.path.isdir(path)):
                #construct full path
                sub_path = []
                for temp in os.listdir(path):
                    sub_path.append(path+temp)
                copy_or_move_files_to(sub_path, dest, flag)
            elif(os.path.isfile(path)):
                #copy file to dest
                ext = os.path.splitext(path)[1]
                if( 'mp3' in ext or 'ogg' in ext ):
                    destfile = dest+os.path.basename(path)
                    if(flag == 'copy'):
                        print "Copying %(src)s to %(dest)s....." % {'src':path, 'dest':destfile}
                        shutil.copy2(path, destfile)
                    elif(flag == 'move'):
                        print "Moving %(src)s to %(dest)s....." % {'src':path, 'dest':destfile}
                        shutil.move(path, destfile)
        else:
            print "Cannot find file or path: %s" % path
            
def helper_get_stor_dir():
    res = api_client.list_all_watched_dirs()
    if(res is None):
        return res
    else:
        if(res['dirs']['1'][-1] != '/'):
            out = res['dirs']['1']+'/'
            return out
        else:
            return res['dirs']['1']

storage_dir = helper_get_stor_dir()
if(storage_dir is None):
    storage_dir = "Unknown" 
else:
    storage_dir += "imported/"
help_text = """
    ========================
    Airtime Import Script
    ========================
    There are two ways to import audio files into Airtime:

    1) Copy or move files into the storage folder

       Copied or moved files will be placed into the folder:
       %s
        
       Files will be automatically organized into the structure
       "Artist/Album/TrackNumber-TrackName-Bitrate.file_extension".

    2) Add a folder to the Airtime library("watch" a folder)
    
       All the files in the watched folder will be imported to Airtime and the
       folder will be monitored to automatically detect any changes. Hence any
       changes done in the folder(add, delete, edit a file) will trigger 
       updates in Airtime libarary.
""" % storage_dir
parser = argparse.ArgumentParser(description=help_text, formatter_class=RawTextHelpFormatter, epilog="  ")#"This script let you do following operations\n- import files\n- add/remove/list watch folders\n- set default storage folder", formatter_class=RawTextHelpFormatter)
# for subcommand move
parser.add_argument('-c','--copy', nargs='+', metavar='FILE', action=CopyAction, help='Copy FILE(s) into the storage directory.\nYou can specify multiple files or directories.')
parser.add_argument('-m','--move', nargs='+', metavar='FILE', action=MoveAction, help='Move FILE(s) into the storage directory.\nYou can specify multiple files or directories.')
parser.add_argument('--watch-add', metavar='DIR', action=WatchAddAction, help='Add DIR to the watched folders list.')
parser.add_argument('--watch-list', nargs=0, action=WatchListAction, help='Show the list of folders that are watched.')
parser.add_argument('--watch-remove', metavar='DIR', action=WatchRemoveAction, help='Remove DIR from the watched folders list.')
parser.add_argument('--storage-dir-set', metavar='DIR', action=StorageSetAction, help='Set storage dir to DIR.')
parser.add_argument('--storage-dir-get', nargs=0, metavar='DIR', action=StorageGetAction, help='Show the current storage dir.')

if('-l' in sys.argv or '--link' in sys.argv):
    print "\nThe [-l][--link] option is deprecated. Please use the --watch-add option.\nTry 'airtime-import -h' for more detail.\n"
else:
    args = parser.parse_args()