class DocumentWriter(object):
"runs a composer"
composer = None
def writestr(self):
if not self.composer:
# default format is XLS
self.composer = ComposerXLS(self.data_model, self.grid, self)
log.info("Setting output format to XLS")
return self.composer.run()
def write(self, filename):
ext = os.path.splitext(filename)[-1].lower()
if ext == "xls":
self.composer = ComposerXLS(self.data_model, self.grid, self)
log.info("Setting output format to XLS, based on file extension")
f = open(filename, "w")
f.write(self.writestr())
f.close()
log.info("wrote file: %s" % filename)
def writestr(self):
if not self.composer:
# default format is XLS
self.composer = ComposerXLS(self.data_model, self.grid, self)
log.info("Setting output format to XLS")
return self.composer.run()
class DocumentWriter(object):
"runs a composer"
composer = None
def writestr(self):
if not self.composer:
# default format is XLS
self.composer = ComposerXLS(self.data_model, self.grid, self)
log.info("Setting output format to XLS")
return self.composer.run()
def write(self, filename):
ext = os.path.splitext(filename)[-1].lower()
if ext == "xls":
self.composer = ComposerXLS(self.data_model, self.grid, self)
log.info("Setting output format to XLS, based on file extension")
f = open(filename, "w")
f.write(self.writestr())
f.close()
log.info("wrote file: %s" % filename)
def write_gdata(self, name, username, password, auth_token=None):
try:
import gdata
import gdata.docs.service
except ImportError:
raise Exception("Must install package 'gdata' to use write_gdata()")
tmp_file, tmp_file_path = tempfile.mkstemp()
self.write(tmp_file_path)
gd_client = gdata.docs.service.DocsService()
gd_client.ssl = True
if not auth_token:
gd_client.ClientLogin(
username,
password,
"marmir-1.0")
else:
#TODO: use the token
raise Exception("oauth not yet supported")
ms = gdata.MediaSource(file_path=tmp_file_path,
content_type='application/vnd.ms-excel')
entry = gd_client.Upload(ms, name)
#cleanup
os.unlink(tmp_file_path)
return gd_client.GetClientLoginToken()
def write(self, filename):
ext = os.path.splitext(filename)[-1].lower()
if ext == "xls":
self.composer = ComposerXLS(self.data_model, self.grid, self)
log.info("Setting output format to XLS, based on file extension")
f = open(filename, "w")
f.write(self.writestr())
f.close()
log.info("wrote file: %s" % filename)
def write(self, filename):
ext = os.path.splitext(filename)[-1].lower()
if ext == "xls":
self.composer = ComposerXLS(self.data_model, self.grid, self)
log.info("Setting output format to XLS, based on file extension")
elif ext == "txt" and pretty_table:
self.composer = ComposerPrettyTable(self.data_model, self.grid,
self)
log.info("Setting output format to TXT, based on file extension")
with open(filename, "wb") as f:
f.write(self.writestr())
log.info("wrote file: %s" % filename)
