# -*- coding: utf-8 -*-
# vi:si:et:sw=4:sts=4:ts=4
##
## Copyright (C) 2011 Async Open Source <http://www.async.com.br>
## All rights reserved
##
## This program is free software; you can redistribute it and/or modify
## it under the terms of the GNU Lesser General Public License as published by
## the Free Software Foundation; either version 2 of the License, or
## (at your option) any later version.
##
## This program is distributed in the hope that it will be useful,
## but WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
## GNU Lesser General Public License for more details.
##
## You should have received a copy of the GNU Lesser General Public License
## along with this program; if not, write to the Free Software
## Foundation, Inc., or visit: http://www.gnu.org/.
##
## Author(s): Stoq Team <stoq-devel@async.com.br>
##
import datetime
import gtk
from stoqlib.api import api
from stoqlib.database.queryexecuter import DateQueryState, DateIntervalQueryState
from stoqlib.domain.person import CallsView, Client, ClientCallsView
from stoqlib.gui.base.dialogs import run_dialog
from stoqlib.gui.editors.callseditor import CallsEditor
from stoqlib.gui.search.searchcolumns import SearchColumn, Column
from stoqlib.gui.search.searcheditor import SearchEditor
from stoqlib.gui.search.searchfilters import DateSearchFilter
from stoqlib.gui.utils.printing import print_report
from stoqlib.lib.translation import stoqlib_gettext
from stoqlib.reporting.callsreport import CallsReport
_ = stoqlib_gettext
[docs]class CallsSearch(SearchEditor):
"""This search can be used directly, to show calls to any kind of person in
the system.
Subclasses can also override I{person_type} to set the person type that
will be available when creating a new call, and I{person_name}, that will be
the column title in this search.
"""
title = _("Calls Search")
search_spec = CallsView
editor_class = CallsEditor
person_type = None
person_name = _('Person')
size = (700, 450)
def __init__(self, store, person=None, date=None, reuse_store=False):
"""
:param store: a store
:param person: If not None, the search will show only call made to
this person.
:param date: If not None, the search will be filtered using this date by
default
:param reuse_store: When False, a new store will be
created/commited when creating a new call. When True, no store
will be created. In this case, I{store} will be utilized.
"""
self.store = store
self.person = person
self._date = date
self._reuse_store = reuse_store
SearchEditor.__init__(self, store)
#
# SearchEditor Hooks
#
[docs] def get_editor_model(self, calls):
return calls.call
[docs] def create_filters(self):
self.set_text_field_columns(['description', 'message'])
self.search.set_query(self.executer_query)
# Date
date_filter = DateSearchFilter(_("Date:"))
self.search.add_filter(date_filter)
self.date_filter = date_filter
if self._date:
date_filter.mode.select_item_by_position(5)
date_filter.start_date.set_date(self._date)
self.search.refresh()
[docs] def get_columns(self):
columns = [Column('date', title=_('Date'),
data_type=datetime.date, width=150, sorted=True),
SearchColumn('description', title=_('Description'),
data_type=str, width=150, expand=True),
SearchColumn('attendant', title=_('Attendant'),
data_type=str, width=100, expand=True)]
if not self.person:
columns.insert(1,
SearchColumn('name', title=self.person_name,
data_type=str, width=150, expand=True))
return columns
[docs] def executer_query(self, store):
client = self.person
date = self.date_filter.get_state()
if isinstance(date, DateQueryState):
date = date.date
elif isinstance(date, DateIntervalQueryState):
date = (date.start, date.end)
else:
date = None
# Use the current connection ('self.store') to show inserted calls,
# before confirm the new person.
return self.search_spec.find_by_client_date(self.store, client, date)
[docs] def run_editor(self, obj):
if self._reuse_store:
self.store.savepoint('before_run_editor_calls')
retval = run_dialog(self.editor_class, self, self.store,
self.store.fetch(obj), self.person,
self.person_type)
if not retval:
self.store.rollback_to_savepoint('before_run_editor_calls')
else:
store = api.new_store()
retval = run_dialog(self.editor_class, self, store,
store.fetch(obj), store.fetch(self.person),
self.person_type)
store.confirm(retval)
store.close()
return retval
#
# Callbacks
#
def _on_results__has_rows(self, widget, has_rows):
self.print_button.set_sensitive(has_rows)
def _on_print_button__clicked(self, widget):
print_report(CallsReport, self.results, list(self.results),
filters=self.search.get_search_filters(),
person=self.person)
[docs]class ClientCallsSearch(CallsSearch):
title = _("Calls Search")
search_spec = ClientCallsView
person_type = Client
person_name = _('Client')