Skip to content

Commit 5ee2c51

Browse files
committed
Some style improvements to the mysql_advanced_data_store.
1 parent 566229e commit 5ee2c51

6 files changed

+163
-122
lines changed

lib/data_store_test.py

+9-6
Original file line numberDiff line numberDiff line change
@@ -1494,8 +1494,9 @@ def _UpdateRandom(self, subjects, fraction, change_test=True):
14941494
elif which == 1:
14951495
# Add another timestamp.
14961496
timestamp_info.append(100 * number_timestamps + 1)
1497-
data_store.DB.Set(subject, self.predicate_template % j, new_value,
1498-
replace=False, timestamp=timestamp_info[-1], token=self.token)
1497+
data_store.DB.Set(
1498+
subject, self.predicate_template % j, new_value,
1499+
replace=False, timestamp=timestamp_info[-1], token=self.token)
14991500
self.values += 1
15001501
self.Register()
15011502
elif which == 2:
@@ -1600,8 +1601,9 @@ def _AddNewSubject(self, client, subjects, i, value, max_attributes=3):
16001601
ts = [100 * (ts + 1) for ts in xrange(number_timestamps)]
16011602
predicates[j] = ts
16021603
values = [(value, t) for t in ts]
1603-
data_store.DB.MultiSet(subject, {self.predicate_template % j: values},
1604-
timestamp=100, replace=False, sync=False, token=self.token)
1604+
data_store.DB.MultiSet(
1605+
subject, {self.predicate_template % j: values},
1606+
timestamp=100, replace=False, sync=False, token=self.token)
16051607
self.Register()
16061608
info = {"name": subject, "attrs": predicates}
16071609
subjects[i] = info
@@ -1635,8 +1637,9 @@ def _AddManyAttributes(self, subjects, many):
16351637
self.values += number_timestamps
16361638
values = [(new_value, t) for t in ts]
16371639
predicates[j] = ts
1638-
data_store.DB.MultiSet(subject, {self.predicate_template % j: values},
1639-
replace=False, timestamp=100, sync=False, token=self.token)
1640+
data_store.DB.MultiSet(
1641+
subject, {self.predicate_template % j: values},
1642+
replace=False, timestamp=100, sync=False, token=self.token)
16401643
self.Register()
16411644
data_store.DB.Flush()
16421645

lib/data_stores/__init__.py

+1
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
# MySql data store not supported.
1919
pass
2020

21+
2122
try:
2223
from grr.lib.data_stores import mysql_advanced_data_store
2324
except ImportError:

0 commit comments

Comments
 (0)