aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/vk/utils/tracking/db/writer
diff options
context:
space:
mode:
Diffstat (limited to 'vk/utils/tracking/db/writer')
-rw-r--r--vk/utils/tracking/db/writer/csv.py21
1 files changed, 12 insertions, 9 deletions
diff --git a/vk/utils/tracking/db/writer/csv.py b/vk/utils/tracking/db/writer/csv.py
index 4594e09..8dc2299 100644
--- a/vk/utils/tracking/db/writer/csv.py
+++ b/vk/utils/tracking/db/writer/csv.py
@@ -3,7 +3,6 @@
# See LICENSE.txt for details.
import csv
-from datetime import datetime
from ..record import Record
@@ -15,25 +14,29 @@ class Writer:
self._fd = open(path, mode)
self._writer = csv.writer(self._fd, lineterminator='\n')
- def _is_valid(self):
+ def __bool__(self):
return self._fd is not None
def __enter__(self):
- if not self._is_valid():
- return None
+ if not self:
+ return self
self._fd.__enter__()
return self
def __exit__(self, *args):
- if self._is_valid():
- self._fd.__exit__(*args)
+ if not self:
+ return
+ self._fd.__exit__(*args)
def flush(self):
- if self._is_valid():
- self._fd.flush()
+ if not self:
+ return
+ self._fd.flush()
def write_record(self, user):
- self._write_row(Record(user).to_list())
+ if not self:
+ return
+ self._write_row(Record.from_user(user).to_row())
self.flush()
def _write_row(self, row):