lib/aipim-rails/parser.rb in aipim-rails-0.0.172 vs lib/aipim-rails/parser.rb in aipim-rails-0.0.174

@@ -5,12 +5,12 @@
 	def self.init(filename)
 		parser(filename)
 	end
 
 	def self.tag_screenshot(line,file,workset)
-		if line && ParserHelper.is_tag?(line) && ParserHelper.is_screenshot?(line)
-			workset[:screenshot] = true
+		if line && ParserHelper.is_tag?(line) 
+			workset[:screenshot] = true if ParserHelper.is_screenshot?(line)
 			line = ParserHelper::read_line(file)
 		end
 		return line, workset
 	end
 
@@ -103,9 +103,14 @@
 		raise "FileNotFound" unless File.exists?(filename)
 		File.open(filename, "r")
 	end
 
 	def self.get_screenshot_files(filename)
-		%x[\ls aipim/screenshots/#{filename}].split("\n")
+		if File.directory?("aipim/screenshots/#{filename}")
+			ret = %x[\ls aipim/screenshots/#{filename}].split("\n")
+		else
+			ret = []
+		end
+		ret
 	end
 
 end