def _on_popup_save__activate(self, menu):
model = self._image_model
name = '%s.png' % model.get_description()
filename = save(current_name=name,
folder=os.path.expanduser('~/'))
if filename:
self._save_image(filename)
def on_ExportBills__activate(self, action):
payments = [
v.payment for v in self.results.get_selected_rows()
if v.method.method_name == 'bill'
]
if not payments:
warning(_('No bill payments were selected'))
return
filename = save(current_name='CNAB.txt',
folder=os.path.expanduser('~/'))
if not filename:
return
bank_number = payments[0].method.destination_account.bank.bank_number
info = get_bank_info_by_number(bank_number)
try:
cnab = info.get_cnab(payments)
except Exception as e:
log.error(''.join(traceback.format_exception(*sys.exc_info())))
warning(_('An error ocurred while generating the CNAB'), str(e))
return
with open(filename, 'w') as fh:
fh.write(cnab)
def save_as_cb(self,w):
#self.session_file=file_browse(gtk.FILE_CHOOSER_ACTION_SAVE)
filename=dialogs.save()
if filename:
filename=self.save_session(filename)
if filename is not None:
self.session_file=filename
self.set_title("Plothole: "+os.path.split(self.session_file)[-1])
def _project_save_as(self, title, project):
filename = save(title, folder=config.lastdirectory)
if not filename:
return False
config.set_lastdirectory(filename)
return self._save(project, filename)
def on_inventory_report__activate(self, *args):
from crisk.reports.invent_report import InventoryReport
filename = save(_('Save report'),
current_name = _('Inventory Report.pdf'))
if filename is not None:
assets = Asset.query().all()
report = InventoryReport(queryset = assets)
report.generate_by(PDFGenerator, filename = filename)
def on_total_vuln_report__activate(self, *args):
from crisk.reports.vuln_report import TotalVulnReport
filename = save(_('Save report'),
current_name=_('Total Vulnerability Risk Report.pdf'))
if filename is not None:
vulns = Vulnerability.query().all()
report = TotalVulnReport(queryset=vulns)
report.generate_by(PDFGenerator, filename)
def on_inventory_report__activate(self, *args):
from crisk.reports.invent_report import InventoryReport
filename = save(_('Save report'),
current_name=_('Inventory Report.pdf'))
if filename is not None:
assets = Asset.query().all()
report = InventoryReport(queryset=assets)
report.generate_by(PDFGenerator, filename=filename)
def _on_popup_save__activate(self, menu):
fname = self.image_model.filename
if '.' in fname:
fname = ''.join(fname.split('.')[:-1])
filename = save(current_name=fname + '.png',
folder=os.path.expanduser('~/'))
if filename:
self._save_image(filename)
def _on_popup_save__activate(self, menu):
fname = self.image_model.filename
if '.' in fname:
fname = ''.join(fname.split('.')[:-1])
filename = save(current_name=fname + '.png',
folder=os.path.expanduser('~/'))
if filename:
self._save_image(filename)
def on_total_vuln_report__activate(self, *args):
from crisk.reports.vuln_report import TotalVulnReport
filename = save(_('Save report'),
current_name = _('Total Vulnerability Risk Report.pdf'))
if filename is not None:
vulns = Vulnerability.query().all()
report = TotalVulnReport(queryset = vulns)
report.generate_by(PDFGenerator, filename)
def on_menuitem_saveas__activate(self, *args):
filename = dialogs.save('Save file as', current_name='untitled.ics')
if filename and not filename.endswith('.ics'):
filename += '.ics'
if filename:
self.mo.save(filename)
self.menuitem_save.set_sensitive(False)
def save_cb(self, w):
if self.session_file is None:
filename=dialogs.save()
self.session_file=self.save_session(filename)
self.set_title("Plothole: "+os.path.split(self.session_file)[-1])
else:
filename=self.save_session(self.session_file)
if filename is not None:
self.session_file=filename
self.set_title("Plothole: "+os.path.split(self.session_file)[-1])
def confirm(self):
start = self.date_filter.get_start_date()
end = self.date_filter.get_end_date()
filename = save(_("Save Sintegra file"),
self.get_toplevel(),
"sintegra-%s.txt" % (start.strftime('%Y-%m'), ))
if not filename:
return
try:
generator = StoqlibSintegraGenerator(self.store, start, end)
generator.write(filename)
except SintegraError, e:
warning(str(e))
return
def _export_product_list(self):
filename = save(_("Save products file"), self.get_toplevel(),
"%s.txt" % (_('products')))
if not filename:
return
tables = [
Sellable,
Join(Storable, Storable.id == Sellable.id),
]
sellables = self.store.using(*tables).find(Sellable)
with open(filename, 'w') as fh:
for sellable in sellables:
# TODO: Add a dialog for the user to choose the format for
# exporting
fh.write('%-20s%s\n' % (sellable.barcode, sellable.description))
def _export_product_list(self):
filename = save(_("Save products file"), self.get_toplevel(),
"%s.txt" % (_('products')))
if not filename:
return
tables = [
Sellable,
Join(Storable, Storable.id == Sellable.id),
]
sellables = self.store.using(*tables).find(Sellable)
with open(filename, 'w') as fh:
for sellable in sellables:
# TODO: Add a dialog for the user to choose the format for
# exporting
fh.write('%-20s%s\n' % (sellable.barcode, sellable.description))
def confirm(self):
start = self.date_filter.get_start_date()
end = self.date_filter.get_end_date()
filename = save(_("Save Sintegra file"), self.get_toplevel(),
"sintegra-%s.txt" % (start.strftime('%Y-%m'), ))
if not filename:
return
try:
generator = StoqlibSintegraGenerator(self.store, start, end)
generator.write(filename)
except SintegraError as e:
warning(str(e))
return
self.close()
def on_new__activate(self, *args):
new_file = save(_('Save'))
if new_file is None:
return None
if not new_file.endswith('.crisk'):
new_file = new_file + '.crisk'
try:
db_url = 'sqlite:///%s' % new_file
metadata.bind = db_url
metadata.bind.echo = False
setup_all()
create_all()
tmp = Basic()
session.commit()
self.db_file = new_file
self.tree.select(self.first)
return True
except Exception, info:
res = error(_("An error has ocurred"), info.__str__())
def on_new__activate(self, *args):
new_file = save(_('Save'))
if new_file is None:
return None
if not new_file.endswith('.crisk'):
new_file = new_file + '.crisk'
try:
db_url = 'sqlite:///%s' % new_file
metadata.bind = db_url
metadata.bind.echo = False
setup_all()
create_all()
tmp = Basic()
session.commit()
self.db_file = new_file
self.tree.select(self.first)
return True
except Exception, info:
res = error(_("An error has ocurred"), info.__str__())
def on_ExportBills__activate(self, action):
payments = [v.payment for v in self.results.get_selected_rows()
if v.method.method_name == 'bill']
if not payments:
warning(_('No bill payments were selected'))
return
filename = save(current_name='CNAB.txt', folder=os.path.expanduser('~/'))
if not filename:
return
bank_number = payments[0].method.destination_account.bank.bank_number
info = get_bank_info_by_number(bank_number)
try:
cnab = info.get_cnab(payments)
except Exception as e:
log.error(''.join(traceback.format_exception(*sys.exc_info())))
warning(_('An error ocurred while generating the CNAB'), str(e))
return
with open(filename, 'w') as fh:
fh.write(cnab)
def export_source_cb(self, w):
filename = dialogs.save()
sel_source=self.source_list.get_selected_rows()[0]
savetxt(filename, transpose(array([sel_source.x_data, sel_source.y_data])), delimiter="\t")
