| """ |
| Extracts the version of the PostgreSQL server. |
| """ |
| |
| import re |
| |
| # This reg-exp is intentionally fairly flexible here. |
| # Needs to be able to handle stuff like: |
| # PostgreSQL 8.3.6 |
| # EnterpriseDB 8.3 |
| # PostgreSQL 8.3 beta4 |
| # PostgreSQL 8.4beta1 |
| VERSION_RE = re.compile(r'\S+ (\d+)\.(\d+)\.?(\d+)?') |
| |
| def _parse_version(text): |
| "Internal parsing method. Factored out for testing purposes." |
| major, major2, minor = VERSION_RE.search(text).groups() |
| try: |
| return int(major), int(major2), int(minor) |
| except (ValueError, TypeError): |
| return int(major), int(major2), None |
| |
| def get_version(cursor): |
| """ |
| Returns a tuple representing the major, minor and revision number of the |
| server. For example, (7, 4, 1) or (8, 3, 4). The revision number will be |
| None in the case of initial releases (e.g., 'PostgreSQL 8.3') or in the |
| case of beta and prereleases ('PostgreSQL 8.4beta1'). |
| """ |
| cursor.execute("SELECT version()") |
| return _parse_version(cursor.fetchone()[0]) |