#!/usr/bin/env python

 __author__ = "Artur Duque de Souza"

 __author_email__ = "artur.souza@indt.org.br"

 __license__ = "GPL"

 __version__ = "0.1"

 import os

 import sys

 import pickle

 import logging

 import lib.utils as utils

 from stat import *

 __all__ = ("FileList", "list_media_files")

 class TranscodedFile(object):

     """This class creates and reads information about transcoded files."""

     opts = {}

     log = logging.getLogger("gms.file_handler")

     def __init__(self, filename, args):

         if filename == "" or not os.path.exists(filename):

             self.opts = args.copy()

             if self.opts["type"][0] != "myth":

                 self.opts["original_mtime"] = os.path.getmtime(

                     self.opts["uri"][0])

             name = os.path.basename(self.opts["uri"][0])

             self.opts["original"] = [name]

             output_file = os.path.basename(self.opts["outfile"][0])

             dat_file = output_file + ".dat";

             dat_path = os.path.join (utils.config.get_transcoded_location(),

                 dat_file);

             output = open(dat_path, "wb")

             # dumps data using the highest protocol

             pickle.dump(self.opts, output, -1)

             output.close()

         else:

             name = os.path.splitext(os.path.basename(filename))[0]

             dat_file = name + ".dat";

             dat_path = os.path.join (utils.config.get_transcoded_location(),

                 dat_file);

             pkl_file = open(dat_path, "rb")

             self.opts = pickle.load(pkl_file)

     # __init__()

 # TranscodedFile

 class FileList(list):

     """Class to hold file's list - reimplements str and repr."""

     def __str__(self):

         ret = ""

         if len(self) > 0:

             for item in self:

                 ret = ret + "%s" % item

         return ret

     # __str__()

     def __repr__(self):

         return self.__str__()

     # __repr__()

 # FileList

 def list_media_files(directory, file_list):

     """Show all the media files with extension defined in the var 'ext'

     that are in the directory, appending each one to 'file_list'."""

     ext = ['mpg', 'avi', 'mp4', 'nuv', 'mpeg', 'mov']

     for root, dirs, files in os.walk(directory):

         for name in files:

             if os.path.splitext(name)[1].strip(".") in ext:

                 dat_file = os.path.join(sys.path[0],root,

                                         os.path.splitext(name)[0]+".dat")

                 if name not in file_list and \

                        os.path.exists(dat_file):

                     file_list.append(name)

     return True

 # list_media_files()