#!/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 = helper_get_stor_dir()+"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 = helper_get_stor_dir()+"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: 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()