33 lines
565 B
Python
33 lines
565 B
Python
# standard imports
|
|
import os
|
|
import sys
|
|
|
|
|
|
class OutputWriter:
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
pass
|
|
|
|
def write(self, k, v):
|
|
raise NotImplementedError()
|
|
|
|
|
|
class StdoutWriter(OutputWriter):
|
|
|
|
def write(self, k, v):
|
|
sys.stdout.write('{}\t{}\n'.format(k, v))
|
|
|
|
|
|
class KVWriter(OutputWriter):
|
|
|
|
def __init__(self, path=None, *args, **kwargs):
|
|
os.stat(path)
|
|
self.path = path
|
|
|
|
|
|
def write(self, k, v):
|
|
fp = os.path.join(self.path, str(k))
|
|
f = open(fp, 'wb')
|
|
f.write(v)
|
|
f.close()
|