|
5 | 5 |
|
6 | 6 | class Pool(object):
|
7 | 7 |
|
8 |
| - ExcludeDBs = ['template0', 'template1', 'postgres'] |
| 8 | + ExcludeDBs = ['template0', 'template1'] |
9 | 9 |
|
10 | 10 | SQL = {
|
11 | 11 | # query type: ( 'if_not_installed', 'if_installed' )
|
@@ -76,21 +76,21 @@ def server_version(self, db=None):
|
76 | 76 | result.decode('ascii'))
|
77 | 77 | return self._cache['server_version']['storage'][db]
|
78 | 78 |
|
79 |
| - def server_version_greater(self, version, db=None, bootstrap=False): |
80 |
| - if not bootstrap: |
81 |
| - db = self._normalize_db(db) |
82 |
| - return self.server_version(db) >= LooseVersion(version) |
83 |
| - else: |
84 |
| - return str( |
85 |
| - self._cache['bootstrap']['version']) >= LooseVersion(version) |
| 79 | + def server_version_greater(self, version, db=None): |
| 80 | + db = self._normalize_db(db) |
| 81 | + return self.server_version(db) >= LooseVersion(version) |
86 | 82 |
|
87 |
| - def server_version_less(self, version, db=None, bootstrap=False): |
88 |
| - if not bootstrap: |
89 |
| - db = self._normalize_db(db) |
90 |
| - return self.server_version(db) <= LooseVersion(version) |
91 |
| - else: |
92 |
| - print(self._cache['bootstrap']['version']) |
93 |
| - return self._cache['bootstrap']['version'] <= LooseVersion(version) |
| 83 | + def server_version_less(self, version, db=None): |
| 84 | + db = self._normalize_db(db) |
| 85 | + return self.server_version(db) <= LooseVersion(version) |
| 86 | + |
| 87 | + def bootstrap_version_greater(self, version): |
| 88 | + return str( |
| 89 | + self._cache['bootstrap']['version']) >= LooseVersion(version) |
| 90 | + |
| 91 | + def bootstrap_version_less(self, version): |
| 92 | + return str( |
| 93 | + self._cache['bootstrap']['version']) <= LooseVersion(version) |
94 | 94 |
|
95 | 95 | def in_recovery(self, db=None):
|
96 | 96 | db = self._normalize_db(db)
|
@@ -159,13 +159,15 @@ def is_pgpro_ee(self, db=None):
|
159 | 159 |
|
160 | 160 | def extension_installed(self, ext, db=None):
|
161 | 161 | db = self._normalize_db(db)
|
162 |
| - result = self.query('select count(*) from pg_catalog.pg_extension\ |
163 |
| - where extname = \'{0}\''.format(ext), db) |
| 162 | + result = self.query( |
| 163 | + 'select count(*) from pg_catalog.pg_extension ' |
| 164 | + 'where extname = \'{0}\''.format(ext), db) |
164 | 165 | return (int(result[0][0])) == 1
|
165 | 166 |
|
166 | 167 | def databases(self):
|
167 |
| - result, databases = self.query('select datname from \ |
168 |
| - pg_catalog.pg_database'), [] |
| 168 | + result, databases = self.query( |
| 169 | + 'select datname from ' |
| 170 | + 'pg_catalog.pg_database'), [] |
169 | 171 | for row in result:
|
170 | 172 | if row[0] not in self.ExcludeDBs:
|
171 | 173 | databases.append(row[0])
|
|
0 commit comments