Mercurial > repos > yufei-luo > s_mart

comparison SMART/Java/Python/mySql/MySqlConnection.py @ 36:44d5973c188c

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Uploaded
author m-zytnicki
date Tue, 30 Apr 2013 15:02:29 -0400
parents
children 169d364ddd91
comparison
equal deleted inserted replaced
35:d94018ca4ada 36:44d5973c188c
1 #
2 # Copyright INRA-URGI 2009-2010
3 #
4 # This software is governed by the CeCILL license under French law and
5 # abiding by the rules of distribution of free software. You can use,
6 # modify and/ or redistribute the software under the terms of the CeCILL
7 # license as circulated by CEA, CNRS and INRIA at the following URL
8 # "http://www.cecill.info".
9 #
10 # As a counterpart to the access to the source code and rights to copy,
11 # modify and redistribute granted by the license, users are provided only
12 # with a limited warranty and the software's author, the holder of the
13 # economic rights, and the successive licensors have only limited
14 # liability.
15 #
16 # In this respect, the user's attention is drawn to the risks associated
17 # with loading, using, modifying and/or developing or reproducing the
18 # software by the user in light of its specific status of free software,
19 # that may mean that it is complicated to manipulate, and that also
20 # therefore means that it is reserved for developers and experienced
21 # professionals having in-depth computer knowledge. Users are therefore
22 # encouraged to load and test the software's suitability as regards their
23 # requirements in conditions enabling the security of their systems and/or
24 # data to be ensured and, more generally, to use and operate it in the
25 # same conditions as regards security.
26 #
27 # The fact that you are presently reading this means that you have had
28 # knowledge of the CeCILL license and that you accept its terms.
29 #
30 #! /usr/bin/env python
31 import os
32 import random
33 import sqlite3
34 from SMART.Java.Python.mySql.MySqlQuery import MySqlQuery
35
36
37 class MySqlConnection(object):
38 """Connection to a database"""
39
40 def __init__(self, verbosity = 0):
41 self.verbosity = verbosity
42 self.databaseName = os.path.join(os.environ.get("SMARTTMPPATH", "."), "smartdb%d" % random.randint(0, 100000))
43 self.connection = sqlite3.connect(self.databaseName)
44 self.executeQuery("PRAGMA journal_mode = OFF")
45 self.executeQuery("PRAGMA synchronous = 0")
46 self.executeQuery("PRAGMA locking_mode = EXCLUSIVE")
47 self.executeQuery("PRAGMA count_change = OFF")
48 self.executeQuery("PRAGMA temp_store = 2")
49
50 def __del__(self):
51 self.connection.close()
52
53
54 def createDatabase(self):
55 pass
56
57
58 def deleteDatabase(self):
59 if os.path.exists(self.databaseName):
60 os.remove(self.databaseName)
61
62
63 def executeQuery(self, command, insertion = False):
64 cursor = self.connection.cursor()
65 query = MySqlQuery(cursor, self.verbosity)
66 try:
67 result = query.execute(command, insertion)
68 self.connection.commit()
69 except:
70 result = query.execute(command, insertion)
71 self.connection.commit()
72 if insertion:
73 return result
74 else:
75 return query
76
77
78 def executeManyQueries(self, commands):
79 cursor = self.connection.cursor()
80 query = MySqlQuery(cursor, self.verbosity)
81 try:
82 for cpt, command in enumerate(commands):
83 query.execute(command)
84 self.connection.commit()
85 except:
86 for cpt, command in enumerate(commands):
87 query.execute(command)
88 self.connection.commit()
89
90
91 def executeManyFormattedQueries(self, command, lines, insertion = False):
92 cursor = self.connection.cursor()
93 query = MySqlQuery(cursor, self.verbosity)
94 for line in lines:
95 result = query.executeFormat(command, line)
96 self.connection.commit()
97 if insertion:
98 return result
99 else:
100 return query
101
102
103 def executeManyQueriesIterator(self, table):
104 cursor = self.connection.cursor()
105 query = MySqlQuery(cursor, self.verbosity)
106 try:
107 for command in table.getIterator():
108 query.execute(command)
109 self.connection.commit()
110 except:
111 for command in table.getIterator():
112 query.execute(command)
113 self.connection.commit()
114
115
116 def executeManyFormattedQueriesIterator(self, table):
117 cursor = self.connection.cursor()
118 query = MySqlQuery(cursor, self.verbosity)
119 try:
120 for command, values in table.getIterator():
121 query.executeFormat(command, values)
122 self.connection.commit()
123 except:
124 for command, values in table.getIterator():
125 query.execute(command, values)
126 self.connection.commit()
127
128
129 def executeFormattedQuery(self, command, parameters, insertion = False):
130 cursor = self.connection.cursor()
131 query = MySqlQuery(cursor, self.verbosity)
132 result = query.executeFormat(command, parameters)
133 self.connection.commit()
134 if insertion:
135 return result
136 else:
137 return query