Example #1
0
	def __init__(self,collections):

		if not os.path.exists('saved_library'): # will change to be name of the .avc file ok
			print("making new save data")
			self.library = Library()
			# for now
			self.library.init_from_xml("test.avc")

		else:
			self.load_library('saved_library',True)

		#self.savedata = open('saved_library','wb')	
		self.index = Index(self.library.get_clip_names())
		self.tag_index = Index(self.library.get_tags())
		self.col_index = Index(collections)
Example #2
0
class Searcher():
	def __init__(self,collections):

		if not os.path.exists('saved_library'): # will change to be name of the .avc file ok
			print("making new save data")
			self.library = Library()
			# for now
			self.library.init_from_xml("test.avc")

		else:
			self.load_library('saved_library',True)

		#self.savedata = open('saved_library','wb')	
		self.index = Index(self.library.get_clip_names())
		self.tag_index = Index(self.library.get_tags())
		self.col_index = Index(collections)

	def load_library(self,filename,debug=False):
		if debug: print("reading old save data")
		with open(filename,'rb') as read:
			try:
				self.library = pickle.load(read)
			except:
				if debug: print('error reading',filename)
				read.close()
				os.remove(filename)
				self.__init__()

	def save_library(self,filename,debug=False):
		if debug: print('saving data')
		with open(filename,'wb') as write:
			pickle.dump(self.library,write)

	def quit(self):
		self.save_library('saved_library',True)

	def search(self,term): # let's try having 2nd value be 
		return self.index.by_prefix(term)

	def search_tag(self,term):
		return self.tag_index.by_prefix(term)

	def search_col(self,term):
		return self.col_index.by_prefix(term)

	def add_tag_to_clip(self,tag,clip):
		clip.add_tag(tag)
		self.library.add_clip_to_tag(clip,tag)
		self.tag_index.add_word(tag)

	def remove_tag_from_clip(self,tag,clip):
		clip.remove_tag(tag)
		self.library.remove_clip_from_tag(clip,tag)

	def remove_tag_from_library(self,tag):
		self.tag_index.remove_word(tag)
		self.library.remove_tag(tag)

	def get_from_name(self,name):
		return self.library.get_clip_from_name(name)

	def has_name(self,name):
		return name in self.library.clips