pp_showlist.py

from glob import glob
import os
import csv
import copy
import json
import string
import ConfigParser
from pp_utils import Monitor

# *************************************
# SHOWLIST CLASS
# ************************************

class ShowList:
    """
    manages a list of show and the show selected from the showlist
    """
    IMAGE_FILES=('Image files', '.gif','.GIF','.jpg','.JPG','.jpeg','.JPEG')
    VIDEO_FILES=('video files','.mp4','.MP4','.mkv','.MKV','.avi','.AVI')
    AUDIO_FILES=('audio files','.mp3','.MP3')

    def __init__(self):
        self.mon=Monitor()
        self.mon.on()
        self.clear()

    def clear(self):
        self._shows = []  #MediaList, stored as a list of dicts
        self._num_shows=0
        self._selected_show_index=-1 # index of currently selected show

    def print_list(self):
        print self._shows

    def length(self):
        return self._num_shows
        
    def at_end(self):
        if self._selected_show_index==self._num_shows-1:
            return True
        else:
            return False
            
    def at_start(self):
        if self._selected_show_index==0:
            return True
        else:
            return False

    def show_is_selected(self):
            if self._selected_show_index>=0:
                return True
            else:
                return False
            
    def selected_show_index(self):
        return self._selected_show_index
        
    def shows(self):
        return self._shows
        
    def show(self,index):
        return self._shows[index]
    
    def selected_show(self):
        """returns a dictionary containing all fields in the selected show """
        return self._selected_show

    def append(self, show_dict):
        """appends a show dictionary to the end of the showlist store"""
        self._shows.append(copy.deepcopy(show_dict))
        self._num_shows+=1

    def copy(self,show_dict,new_name):
        show_to_copy=copy.deepcopy(show_dict)
        show_to_copy['show-ref']=new_name
        show_to_copy['title']=new_name
        return show_to_copy

    

    def remove(self,index):
        self._shows.pop(index)
        self._num_shows-=1
        # deselect any show, saves worrying about whether index needs changing
        self._selected_show_index=-1

    def replace(self,index,replacement):
        self._shows[index]= replacement
            
    def first(self):
        self.select(0)

    def next(self):
        if self.length()>0:
            if self._selected_show_index== self.length()-1:
                index=0
            else:
                index= self._selected_show_index+1
            self.select(index)

    def previous(self):
        if self.length()>0:
            if self._selected_show_index == 0:
                index=self.length()-1
            else:
                index= self._selected_show_index-1
            self.select(index)
    
    def index_of_show(self,wanted_show):
        index = 0
        for show in self._shows:
            if show['show-ref']==wanted_show:
                return index
            index +=1
        return -1

    def select(self, index):
        """
        user clicks on a show in the Shows list so try and select it.
        """
        # needs forgiving int for possible tkinter upgrade
        self._selected_show_index=index
        self._selected_show=self._shows[index]

        
    def open_json(self,filename):
        """
        opens a saved showlistlist
        showlists are stored in files as json arrays within a object 'shows'.
        shows are stored internally as a list of dictionaries in self._shows
        """
        if filename !="" and os.path.exists(filename):
            ifile  = open(filename, 'rb')
            sdict= json.load(ifile)
            ifile.close()
            self._shows=sdict['shows']
            if 'issue' in sdict:
                self.issue= sdict['issue']
            else:
                self.issue="1.0"
            self._num_shows=len(self._shows)
            self._selected_show_index=-1
            return True
        else:
            return False

    def sissue(self):
        return self.issue
        
            
    def save_list(self,filename):
        """ save a showlist """
        if filename=="":
            return False
        if os.name=='nt':
            filename = string.replace(filename,'/','\\')
        else:
            filename = string.replace(filename,'\\','/')
        dic={'issue':self.issue,'shows':self._shows}
        ofile  = open(filename, "wb")
        json.dump(dic,ofile,sort_keys=True,indent=1)
        ofile.close()
        return
            
# =====================================================
# old stuff
    def open_cfg(self,filename):
        """
        opens a saved showlist
        each show is a section of a confiParser file
        shows are stored as a list of dictionaries in self._shows
        """
        if filename !="" and os.path.exists(filename):
            self.config=ConfigParser.ConfigParser()
            self.config.read(filename)
            self._shows=[]
            for section in self.sections():
                self._shows.append(self.dictionary_of(section))
            self._num_shows=len(self._shows)
            self._selected_show_index=-1
            return True
        else:
            return False
    def has_show(self,section):
        return self.config.has_section(section)

    def sections(self):
        return self.config.sections()
    
    def get(self,section,item):
        return self.config.get(section,item,0)

    def dictionary_of(self,section):
        return dict(self.config.items(section))







# **************
# Test Harness
# *************

if __name__ == '__main__':
    # make form a directory of files
    ml=MediaList()
    ml.make_list_from_dir("/home/pi/pipresents/media")
    #ml.print_list
    ml.save_list("/home/pi/pipresents/temp/test_ml.json")
    
    # make from a csv file. Fields - location,title,type
    ml=MediaList()
    ml.open_csv("/home/pi/pipresents/pp_profiles/pp_profile/images.csv")
    #ml.print_list
    ml.save_list("/home/pi/pipresents/temp/test_mlccsv.json")