1 # copyright 2003-2010 LOGILAB S.A. (Paris, FRANCE), all rights reserved.

     2 # contact http://www.logilab.fr/ -- mailto:contact@logilab.fr

     3 #

     4 # This file is part of CubicWeb.

     5 #

     6 # CubicWeb is free software: you can redistribute it and/or modify it under the

     7 # terms of the GNU Lesser General Public License as published by the Free

     8 # Software Foundation, either version 2.1 of the License, or (at your option)

     9 # any later version.

    10 #

    11 # CubicWeb is distributed in the hope that it will be useful, but WITHOUT

    12 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS

    13 # FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more

    14 # details.

    15 #

    16 # You should have received a copy of the GNU Lesser General Public License along

    17 # with CubicWeb.  If not, see <http://www.gnu.org/licenses/>.

    18 import webbrowser

    19 reload(webbrowser)

    20 

    21 from sensor.Sensor import Sensor

    22 from utils import datatypes, i18n

    23 

    24 from cubicweb.dbapi import connect

    25 

    26 _ = str

    27 

    28 class RQLSensor(Sensor):

    29 

    30     def __init__(self, *args):

    31         global _; _ = i18n.Translator("rql-desklet")

    32         Sensor.__init__(self)

    33         # define configuration

    34         self._set_config_type("appid", datatypes.TYPE_STRING, "")

    35         self._set_config_type("user", datatypes.TYPE_STRING, "")

    36         self._set_config_type("passwd", datatypes.TYPE_SECRET_STRING, "")

    37         self._set_config_type("rql", datatypes.TYPE_STRING, "")

    38         self._set_config_type("url", datatypes.TYPE_STRING, "")

    39         self._set_config_type("delay", datatypes.TYPE_STRING, "600")

    40         # default timer

    41         self._add_timer(20, self.__update)

    42 

    43     def get_configurator(self):

    44         configurator = self._new_configurator()

    45         configurator.set_name(_("RQL"))

    46         configurator.add_title(_("CubicWeb source settings"))

    47         configurator.add_entry(_("ID",), "appid", _("The application id of this source"))

    48         configurator.add_entry(_("User",), "user", _("The user to connect to this source"))

    49         configurator.add_entry(_("Password",), "passwd", _("The user's password to connect to this source"))

    50         configurator.add_entry(_("URL",), "url", _("The url of the web interface for this source"))

    51         configurator.add_entry(_("RQL",), "rql", _("The rql query"))

    52         configurator.add_entry(_("Update interval",), "delay", _("Delay in seconds between updates"))

    53         return configurator

    54 

    55 

    56     def call_action(self, action, path, args=[]):

    57         index = path[-1]

    58         output = self._new_output()

    59         if action=="enter-line":

    60             # change background

    61             output.set('resultbg[%s]' % index, 'yellow')

    62         elif action=="leave-line":

    63             # change background

    64             output.set('resultbg[%s]' % index, 'black')

    65         elif action=="click-line":

    66             # open url

    67             output.set('resultbg[%s]' % index, 'black')

    68             webbrowser.open(self._urls[index])

    69         self._send_output(output)

    70 

    71     def __get_connection(self):

    72         try:

    73             return self._v_cnx

    74         except AttributeError:

    75             appid, user, passwd = self._get_config("appid"), self._get_config("user"), self._get_config("passwd")

    76             cnx = connect(database=appid, login=user, password=passwd)

    77             self._v_cnx = cnx

    78             return cnx

    79 

    80     def __run_query(self, output):

    81         base = self._get_config('url')

    82         rql = self._get_config('rql')

    83         cnx = self.__get_connection()

    84         cursor = cnx.cursor()

    85         try:

    86             rset = cursor.execute(rql)

    87         except Exception:

    88             del self._v_cnx

    89             raise

    90         self._urls = []

    91         output.set('layout', 'vertical, 14')

    92         output.set('length', rset.rowcount)

    93         i = 0

    94         for line in rset:

    95             output.set('result[%s]' % i, ', '.join([str(v) for v in line[1:]]))

    96             output.set('resultbg[%s]' % i, 'black')

    97             try:

    98                 self._urls.append(base % 'Any X WHERE X eid %s' % line[0])

    99             except Exception:

   100                 self._urls.append('')

   101             i += 1

   102 

   103     def __update(self):

   104         output = self._new_output()

   105         try:

   106             self.__run_query(output)

   107         except Exception as ex:

   108             import traceback

   109             traceback.print_exc()

   110             output.set('layout', 'vertical, 10')

   111             output.set('length', 1)

   112             output.set('result[0]', str(ex))

   113         self._send_output(output)

   114         self._add_timer(int(self._get_config('delay'))*1000, self.__update)

   115 

   116 

   117 def new_sensor(args):

   118     return RQLSensor(*args)