continue
movie['year'] = int(vals[0])
movies.append(movie)
return movies
if __name__ == '__main__':
log = open('/home/thayes/Projects/FiLTH/logs/MovieTaggerTest.log', 'w')
tagger = MovieTagger('/home/thayes/Projects/FiLTH/sql/tag_given_to.sql', '/home/thayes/Projects/FiLTH/sql/tag.sql', log)
#movies = initMovies(100)
try:
#for m in movies:
# tagger.promptUserForTag(m['mid'], m['title'], m['year'])
tagger.promptUserForTag(999999, 'Foo', 1999)
except (QuitException, KeyboardInterrupt):
print '\nQUITTING\n'
finally:
if tagger.hasInserts():
while True:
response = raw_input('\nThere are still unwritten sql insert statements. Write them out? ').lower()
if response not in ['y','n']:
print "Only 'y'/'n'\n"
continue
if response == 'y':
tagger.writeTagInsertsToFile(open('/home/thayes/Projects/FiLTH/sql/tag.sql', 'a'))
tagger.writeTagGivenToInsertsToFile(open('/home/thayes/Projects/FiLTH/sql/tag_given_to.sql', 'a'))
break
else:
break
tagger.close()
sys.stderr.write("**ERROR: opening file: " + str(e) + ".\n")
sys.exit()
movieTagger = MovieTagger(TAG_GIVEN_TO_FILENAME, TAG_FILENAME, logger)
lastProcessed = tempFile.read()
log('main', 'last mid processed (read from ' + TEMP_FILENAME + '): ' + lastProcessed)
lastProcessed = int(lastProcessed)
#grab all movies seen from movies file
initMovies(lastProcessed)
try:
for movie in movies:
movieTagger.promptUserForTag(movie['mid'], movie['title'], movie['year'])
lastProcessed += 1
except QuitException, KeyboardInterrupt:
if movieTagger.hasInserts():
while True:
response = raw_input('\n**WARNING: There are still unwritten sql insert statements. Write them out? ').lower()
if response not in ['y','n']:
print "Only 'y'/'n'\n"
continue
if response == 'y':
movieTagger.writeTagInsertsToFile(tagFile)
movieTagger.writeTagGivenToInsertsToFile(tagGivenToFile)
quit(lastProcessed)
break
finally:
closeFiles()
