forum_modules/pgfulltext/startup.py

   1 import os\r
   2 from forum.models import KeyValue\r
   3 from django.db import connection, transaction\r
   4 \r
   5 KEY = 'PG_FTSTRIGGERS_VERSION'\r
   6 VERSION = 3\r
   7 install = False\r
   8 \r
   9 try:\r
  10     version = KeyValue.objects.get(key=KEY).value\r
  11     if version < VERSION:\r
  12         install = True\r
  13 except:\r
  14     install = True\r
  15 \r
  16 \r
  17 if install:\r
  18     f = open(os.path.join(os.path.dirname(__file__), 'pg_fts_install.sql'), 'r')\r
  19 \r
  20     try:\r
  21         cursor = connection.cursor()\r
  22         cursor.execute(f.read())\r
  23         transaction.commit_unless_managed()\r
  24 \r
  25         try:\r
  26             kv = KeyValue.objects.get(key=KEY)\r
  27         except:\r
  28             kv = KeyValue(key=KEY)\r
  29 \r
  30         kv.value = VERSION\r
  31         kv.save()\r
  32         \r
  33     except Exception, e:\r
  34         #import sys, traceback\r
  35         #traceback.print_exc(file=sys.stdout)\r
  36         pass\r
  37     finally:\r
  38         cursor.close()\r
  39 \r
  40     f.close()\r