aboutsummaryrefslogtreecommitdiffstats
path: root/libbe/storage
diff options
context:
space:
mode:
Diffstat (limited to 'libbe/storage')
-rw-r--r--libbe/storage/__init__.py14
-rw-r--r--libbe/storage/base.py5
-rw-r--r--libbe/storage/util/upgrade.py24
-rw-r--r--libbe/storage/vcs/base.py14
4 files changed, 33 insertions, 24 deletions
diff --git a/libbe/storage/__init__.py b/libbe/storage/__init__.py
index c58ec34..104b1e1 100644
--- a/libbe/storage/__init__.py
+++ b/libbe/storage/__init__.py
@@ -10,6 +10,17 @@ NotWriteable = base.NotWriteable
NotReadable = base.NotReadable
EmptyCommit = base.EmptyCommit
+# a list of all past versions
+STORAGE_VERSIONS = ['Bugs Everywhere Tree 1 0',
+ 'Bugs Everywhere Directory v1.1',
+ 'Bugs Everywhere Directory v1.2',
+ 'Bugs Everywhere Directory v1.3',
+ 'Bugs Everywhere Directory v1.4',
+ ]
+
+# the current version
+STORAGE_VERSION = STORAGE_VERSIONS[-1]
+
def get_storage(location):
"""
Return a Storage instance from a repo location string.
@@ -21,4 +32,5 @@ def get_storage(location):
__all__ = [ConnectionError, InvalidID, InvalidRevision,
InvalidDirectory, NotWriteable, NotReadable,
- EmptyCommit, get_storage]
+ EmptyCommit, STORAGE_VERSIONS, STORAGE_VERSION,
+ get_storage]
diff --git a/libbe/storage/base.py b/libbe/storage/base.py
index 97c8b29..f32353f 100644
--- a/libbe/storage/base.py
+++ b/libbe/storage/base.py
@@ -10,6 +10,7 @@ import pickle
import types
from libbe.error import NotSupported
+import libbe.storage
from libbe.util.tree import Tree
from libbe.util import InvalidObject
from libbe import TESTING
@@ -133,6 +134,10 @@ class Storage (object):
"""Return a version string for this backend."""
return '0'
+ def storage_version(self):
+ """Return the storage format for this backend."""
+ return libbe.storage.STORAGE_VERSION
+
def is_readable(self):
return self.readable and self._readable
diff --git a/libbe/storage/util/upgrade.py b/libbe/storage/util/upgrade.py
index ce6831d..20ef1e4 100644
--- a/libbe/storage/util/upgrade.py
+++ b/libbe/storage/util/upgrade.py
@@ -25,6 +25,7 @@ import sys
import libbe
import libbe.bug
import libbe.storage.util.mapfile as mapfile
+from libbe.storage import STORAGE_VERSIONS, STORAGE_VERSION
#import libbe.storage.vcs # delay import to avoid cyclic dependency
import libbe.ui.util.editor
import libbe.util
@@ -32,17 +33,6 @@ import libbe.util.encoding as encoding
import libbe.util.id
-# a list of all past versions
-BUGDIR_DISK_VERSIONS = ['Bugs Everywhere Tree 1 0',
- 'Bugs Everywhere Directory v1.1',
- 'Bugs Everywhere Directory v1.2',
- 'Bugs Everywhere Directory v1.3',
- 'Bugs Everywhere Directory v1.4',
- ]
-
-# the current version
-BUGDIR_DISK_VERSION = BUGDIR_DISK_VERSIONS[-1]
-
class Upgrader (object):
"Class for converting between different on-disk BE storage formats."
initial_version = None
@@ -304,16 +294,16 @@ for upgrader in upgraders:
upgrade_classes[(upgrader.initial_version,upgrader.final_version)]=upgrader
def upgrade(path, current_version,
- target_version=BUGDIR_DISK_VERSION):
+ target_version=STORAGE_VERSION):
"""
Call the appropriate upgrade function to convert current_version
to target_version. If a direct conversion function does not exist,
use consecutive conversion functions.
"""
- if current_version not in BUGDIR_DISK_VERSIONS:
+ if current_version not in STORAGE_VERSIONS:
raise NotImplementedError, \
"Cannot handle version '%s' yet." % current_version
- if target_version not in BUGDIR_DISK_VERSIONS:
+ if target_version not in STORAGE_VERSIONS:
raise NotImplementedError, \
"Cannot handle version '%s' yet." % current_version
@@ -324,10 +314,10 @@ def upgrade(path, current_version,
u.upgrade()
else:
# consecutive single-step conversion
- i = BUGDIR_DISK_VERSIONS.index(current_version)
+ i = STORAGE_VERSIONS.index(current_version)
while True:
- version_a = BUGDIR_DISK_VERSIONS[i]
- version_b = BUGDIR_DISK_VERSIONS[i+1]
+ version_a = STORAGE_VERSIONS[i]
+ version_b = STORAGE_VERSIONS[i+1]
try:
upgrade_class = upgrade_classes[(version_a, version_b)]
except KeyError:
diff --git a/libbe/storage/vcs/base.py b/libbe/storage/vcs/base.py
index a45f1fe..1df08cf 100644
--- a/libbe/storage/vcs/base.py
+++ b/libbe/storage/vcs/base.py
@@ -34,6 +34,7 @@ import sys
import tempfile
import libbe
+import libbe.storage
import libbe.storage.base
import libbe.util.encoding
from libbe.storage.base import EmptyCommit, InvalidRevision
@@ -652,7 +653,7 @@ os.listdir(self.get_path("bugs")):
if not os.path.isdir(self.be_dir):
raise libbe.storage.base.ConnectionError(self)
self._cached_path_id.connect()
- self.check_disk_version()
+ self.check_storage_version()
def disconnect(self):
self._cached_path_id.disconnect()
@@ -879,18 +880,19 @@ os.listdir(self.get_path("bugs")):
f.close()
return (summary, body)
- def check_disk_version(self):
- version = self.disk_version()
- if version != upgrade.BUGDIR_DISK_VERSION:
+ def check_storage_version(self):
+ version = self.storage_version()
+ if version != libbe.storage.STORAGE_VERSION:
upgrade.upgrade(self.repo, version)
- def disk_version(self, path=None):
+ def storage_version(self, path=None):
"""
Requires disk access.
"""
if path == None:
path = os.path.join(self.repo, '.be', 'version')
- return libbe.util.encoding.get_file_contents(path).rstrip('\n')
+ return libbe.util.encoding.get_file_contents(
+ path, decode=True).rstrip('\n')