| # |
| # |
| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| # |
| # |
| # A handle object for convenience in opening a svn repository |
| |
| import sys |
| |
| # We need a bsddb linked to the same version of Berkeley DB as Subversion is |
| try: |
| import bsddb3 as bsddb |
| except ImportError: |
| import bsddb |
| |
| # Publish the result |
| sys.modules['svnfs_bsddb'] = bsddb |
| |
| from svnfs_bsddb.db import * |
| |
| class Ctx: |
| def __init__(self, dbhome, readonly=None): |
| self.env = self.uuids_db = self.revs_db = self.txns_db = self.changes_db \ |
| = self.copies_db = self.nodes_db = self.reps_db = self.strings_db = \ |
| None |
| try: |
| self.env = DBEnv() |
| self.env.set_lk_detect(DB_LOCK_RANDOM) |
| self.env.set_get_returns_none(1) |
| self.env.open(dbhome, DB_CREATE | DB_INIT_MPOOL | DB_INIT_TXN \ |
| | DB_INIT_LOCK | DB_INIT_LOG) |
| def open_db(dbname): |
| db = DB(self.env) |
| dbflags = 0 |
| if readonly: |
| dbflags = DB_RDONLY |
| db.open(dbname, flags=dbflags) |
| return db |
| self.uuids_db = open_db('uuids') |
| self.revs_db = open_db('revisions') |
| self.txns_db = open_db('transactions') |
| self.changes_db = open_db('changes') |
| self.copies_db = open_db('copies') |
| self.nodes_db = open_db('nodes') |
| self.reps_db = open_db('representations') |
| self.strings_db = open_db('strings') |
| except: |
| self.close() |
| raise |
| |
| def close(self): |
| def close_if_not_None(i): |
| if i is not None: |
| i.close() |
| close_if_not_None(self.uuids_db ) |
| close_if_not_None(self.revs_db ) |
| close_if_not_None(self.txns_db ) |
| close_if_not_None(self.changes_db ) |
| close_if_not_None(self.copies_db ) |
| close_if_not_None(self.nodes_db ) |
| close_if_not_None(self.reps_db ) |
| close_if_not_None(self.strings_db ) |
| close_if_not_None(self.env ) |
| self.env = self.uuids_db = self.revs_db = self.txns_db = self.changes_db \ |
| = self.copies_db = self.nodes_db = self.reps_db = self.strings_db = \ |
| None |
| |
| # And now, some utility functions |
| def get_whole_string(self, key): |
| cur = self.strings_db.cursor() |
| try: |
| rec = cur.set(key) |
| if rec is None: |
| raise DBNotFoundError |
| str = "" |
| while rec: |
| str = str + (rec[1] or "") |
| rec = cur.next_dup() |
| finally: |
| cur.close() |
| return str |
| |