54 lines
1.1 KiB
Python
54 lines
1.1 KiB
Python
# standard imports
|
||
import os
|
||
import sys
|
||
import logging
|
||
import urllib.request
|
||
|
||
logg = logging.getLogger(__name__)
|
||
|
||
|
||
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):
|
||
try:
|
||
os.stat(path)
|
||
except FileNotFoundError:
|
||
os.makedirs(path)
|
||
self.path = path
|
||
|
||
|
||
def write(self, k, v):
|
||
fp = os.path.join(self.path, str(k))
|
||
logg.debug('path write {} {}'.format(fp, str(v)))
|
||
f = open(fp, 'wb')
|
||
f.write(v)
|
||
f.close()
|
||
|
||
|
||
class HTTPWriter(OutputWriter):
|
||
|
||
def __init__(self, path=None, *args, **kwargs):
|
||
super(HTTPWriter, self).__init__(*args, **kwargs)
|
||
self.path = path
|
||
|
||
|
||
def write(self, k, v):
|
||
rq = urllib.request.Request(self.path, method='POST', data=v)
|
||
r = urllib.request.urlopen(rq)
|
||
logg.info('proof submited at {}'.format(r.read))
|