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

# Copyright 2006-2007 (C) Raster Software Vigo (Sergio Costas)
# Copyright 2006-2007 (C) Peter Gill - win32 parts

# This file is part of DeVeDe
#
# DeVeDe is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#
# DeVeDe is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.


import os
import shutil
import glob

class delete_files:

	""" This class allows to delete temporary files and directories """
	
	def __init__(self,filename,filefolder):
		
		self.filename=filename
		self.filefolder=filefolder
		self.filepath=filefolder+filename


	def delete_all(self):
		
		self.delete_avi()
		self.delete_bin_cue()
		self.delete_directory()
		self.delete_iso()
		self.delete_menu_temp()
		self.delete_menu()
		self.delete_mpg()
		self.delete_sub_xml()
		self.delete_xml()


	def delete_mpg(self):

		print "Deleting "+self.filename+"_??_??.mpg"
		try:
			listfiles=glob.glob(self.filepath+"_[0-9][0-9]_[0-9][0-9]"+".mpg")
			for afile in listfiles:
				print "Trying to delete "+afile
				try:
					os.remove(afile)
				except OSError:
					print str(afile.strip())+" not found to remove"
		except OSError:
			print self.filepath+"._??_??.mpg not found"
	
			
	def delete_avi(self):

		try:
			listfiles=glob.glob(self.filepath+"_01_[0-9][0-9]"+".avi")
			for afile in listfiles:
				try:
					os.remove(afile)
				except OSError:
					print str(afile.strip())+" not found to remove"
		except OSError:
			print self.filepath+"._01_??.avi not found"
			
	
	def delete_xml(self):
		
		try:
			os.remove(self.filepath+".xml")
		except OSError:
			print self.filepath+".xml not found"


	def delete_menu_temp(self):
	
		try:
			os.remove(self.filepath+"_menu.xml")
		except OSError:
			print self.filepath+"_menu.xml not found"
		try:
			os.remove(self.filepath+"_menu.mpg")
		except OSError:
			print self.filepath+"_menu.mpg not found"
		try:
			os.remove(self.filepath+"_menu_bg.png")
		except OSError:
			print self.filepath+"_menu_bg.png not found"
		try:
			os.remove(self.filepath+"_menu_bg_active.png")
		except OSError:
			print self.filepath+"_menu_bg_active.png not found"
		try:
			os.remove(self.filepath+"_menu_bg_active_out.png")
		except OSError:
			print self.filepath+"_menu_bg_active_out.png not found"


	def delete_menu(self):
		try:
			os.remove(self.filepath+"_menu2.mpg")
		except OSError:
			print self.filepath+"_menu2.mpg not found"


	def delete_sub_xml(self):

		try:
			os.remove(self.filepath+"_sub.xml")
		except OSError:
			print self.filepath+".xml not found"
		
	
	def delete_iso(self):
		
		try:
			os.remove(self.filepath+".iso")
		except OSError:
			print self.filepath+".iso not found"


	def delete_bin_cue(self):
		
		try:
			os.remove(self.filepath+".cue")
		except OSError:
			print self.filepath+".cue not found"
			
		try:
			os.remove(self.filepath+".bin")
		except OSError:
			print self.filepath+".bin not found"


	def delete_directory(self):

		try:
			shutil.rmtree(self.filepath)
		except OSError:
			print self.filepath+os.sep+" not found"
