diff --git a/rpm/item_import.py b/rpm/item_import.py new file mode 100644 index 0000000..a19ce9f --- /dev/null +++ b/rpm/item_import.py @@ -0,0 +1,35 @@ +#simple script for importing u_iem + +import redis +import sys + +from config import Config + + +#config +cfg = Config() + +#redis connect (pipeline) +rC = redis.Redis(cfg.g('redis','host')) +pL = rC.pipeline(cfg.g('redis','port')) + + +#read file +itemfile = "/Users/elliottcordo/Projects/Caserta/rpM/sample_data/movie_lens/u_item.txt" # str(sys.argv) +items = open (itemfile) + +#read the lines +records = items.readlines() + +#loop split and write to redis +for line in records: + #print line + parsed_line = line.split('|') + #print parsed_line[0]+ ' name ' + parsed_line[1] + pL.hset ('I'+str(parsed_line[0]),'name', parsed_line[1]) + pL.hset ('I'+str(parsed_line[0]),'imdb', parsed_line[4]) + + +pL.execute() + +items.close()