def testStoreClientEntry(self):
currTime = time.time()
self.assertEqual(get_client_stats_db().find().count(), 0)
stats.storeClientEntry("testuser", "testfield", currTime, 0.002, {"metadata_key": "metadata_val"})
self.assertEqual(get_client_stats_db().find().count(), 1)
self.assertEqual(get_client_stats_db().find({"user": "******"}).count(), 1)
self.assertEqual(get_client_stats_db().find({"ts": currTime}).count(), 1)
def testStoreClientEntry(self):
currTime = time.time()
self.assertEqual(get_client_stats_db().query_tags(), None)
success = stats.storeClientEntry("testuser", "testfield", currTime, 0.002, {"metadata_key": "metadata_val"})
savedEntries = get_client_stats_db().query_tags()
self.assertEquals(type(savedEntries), list)
self.assertEquals(len(savedEntries), 1)
entry = savedEntries[0]
self.assertEquals(entry["Metadata"]["metadata_key"], "metadata_val")
def storeClientEntries(fname):
df = pandas.read_csv(fname)
for c in range(len(df)):
try:
user = df['user'][c]
stat = df['stat'][c]
ts = int(float(df['client_ts'][c]))
reading = df['reading'][c]
if reading.lower() == 'none':
reading = 1.0
else:
reading = float(reading)
metadata = {}
for key in df:
if key not in ['user', 'stat', 'client_ts', 'reading']:
metadata[key] = df[key][c]
storeClientEntry(user, stat, ts, reading, metadata)
except Exception as e:
print(e)
#print(df['reading'][c] == 'None')
print(df['reading'][c] == 'None')
def testStoreClientEntry(self):
currTime = time.time()
self.assertEqual(get_client_stats_db().query_tags(), None)
success = stats.storeClientEntry("testuser", "testfield", currTime,
0.002,
{"metadata_key": "metadata_val"})
savedEntries = get_client_stats_db().query_tags()
self.assertEquals(type(savedEntries), list)
self.assertEquals(len(savedEntries), 1)
entry = savedEntries[0]
self.assertEquals(entry["Metadata"]["metadata_key"], "metadata_val")
def storeClientEntries(fname):
df = pandas.read_csv(fname)
for c in range(len(df)):
try:
user = df['user'][c]
stat = df['stat'][c]
ts = int(float(df['client_ts'][c]))
reading = df['reading'][c]
if reading.lower() == 'none':
reading = 1.0
else:
reading = float(reading)
metadata = {}
for key in df:
if key not in ['user', 'stat', 'client_ts', 'reading']:
metadata[key] = df[key][c]
storeClientEntry(user, stat, ts, reading, metadata)
except Exception as e:
print(e)
#print(df['reading'][c] == 'None')
print(df['reading'][c] == 'None')
