import urllib2
from redis_completion_2 import RedisEngine
engine = RedisEngine(prefix='stocks')
def load_data():
url = 'http://media.charlesleifer.com/downloads/misc/NYSE.txt'
contents = urllib2.urlopen(url).read()
for row in contents.splitlines()[1:]:
ticker, company = row.split('\t')
engine.store_json(ticker, company, {'ticker': ticker, 'company': company}) # id, search phrase, data
def search(p, **kwargs):
return engine.search_json(p, **kwargs)
if __name__ == '__main__':
engine.flush()
print 'Loading data (may take a few seconds...)'
#load_data()
engine.store_json(54548, 'test company omar', {'ticker': 54548, 'company': 'test company omar'})
engine.store_json(54549, 'test company', {'ticker': 54548, 'company': 'test company'})
engine.store_json(54550, 'test company aaaaa', {'ticker': 54548, 'company': 'test company aaaaa'})
engine.store_json(54551, 'test company 11a', {'ticker': 54548, 'company': 'test company 11a'})
print 'Search data by typing a partial phrase, like "uni sta"'
print 'Type "q" at any time to quit'
print (search('t'))
