1
0
Fork 0
mirror of https://github.com/borgbackup/borg.git synced 2024-12-25 09:19:31 +00:00

Added "init" subcommand

This commit is contained in:
Jonas Borgström 2010-10-26 21:48:43 +02:00
parent e5fad313f7
commit e650e195e1
3 changed files with 27 additions and 14 deletions

View file

@ -18,6 +18,10 @@ def open_store(self, location):
def exit_code_from_logger(self):
return 1 if self.level_filter.count.get('ERROR') else 0
def do_init(self, args):
Store(args.store.path, create=True)
return self.exit_code_from_logger()
def do_create(self, args):
store = self.open_store(args.archive)
keychain = KeyChain(args.keychain)
@ -115,6 +119,12 @@ def run(self, args=None):
subparser = subsubparsers.add_parser('change-password')
subparser.set_defaults(func=self.do_keychain_chpass)
subparser = subparsers.add_parser('init')
subparser.set_defaults(func=self.do_init)
subparser.add_argument('store', metavar='STORE',
type=location_validator(archive=False),
help='Store to initialize')
subparser = subparsers.add_parser('create')
subparser.set_defaults(func=self.do_create)
subparser.add_argument('archive', metavar='ARCHIVE',

View file

@ -23,16 +23,24 @@ class AlreadyExists(KeyError):
ACTIVE = 'Active'
BAND_LIMIT = 1 * 1024 * 1024
def __init__(self, path):
def __init__(self, path, create=False):
self.read_fd = None
self.write_fd = None
if not os.path.exists(path):
if create:
self.create(path)
self.open(path)
def get_option(self, key):
return self.cursor.execute('SELECT value FROM system WHERE key=?', (key,)) \
.fetchone()[0]
def set_option(self, key, value):
return self.cursor.execute('UPDATE system SET value=? WHERE key=?',
(value, key))
def open(self, path):
if not os.path.isdir(path):
raise Exception('%s Does not look like a store')
raise Exception('%s Does not look like a store' % path)
db_path = os.path.join(path, 'dedupestore.db')
if not os.path.exists(db_path):
raise Exception('%s Does not look like a store2')
@ -44,15 +52,6 @@ def open(self, path):
self.cursor = self.cnx.cursor()
self._begin()
def get_option(self, key):
return self.cursor.execute('SELECT value FROM system WHERE key=?', (key,)) \
.fetchone()[0]
def set_option(self, key, value):
return self.cursor.execute('UPDATE system SET value=? WHERE key=?',
(value, key))
def _begin(self):
if self.read_fd:
self.read_fd.close()
@ -76,7 +75,10 @@ def _begin(self):
band += 1
def create(self, path):
os.mkdir(path)
if os.path.exists(path) and (not os.path.isdir(path) or os.listdir(path)):
raise Exception('Path "%s" already exists' % path)
if not os.path.exists(path):
os.mkdir(path)
os.mkdir(os.path.join(path, 'bands'))
cnx = sqlite3.connect(os.path.join(path, 'dedupestore.db'))
cnx.execute('CREATE TABLE objects(ns BINARY NOT NULL, id BINARY NOT NULL, '
@ -212,7 +214,7 @@ class StoreTestCase(unittest.TestCase):
def setUp(self):
self.tmppath = tempfile.mkdtemp()
self.store = Store(os.path.join(self.tmppath, 'store'))
self.store = Store(os.path.join(self.tmppath, 'store'), create=True)
def tearDown(self):
shutil.rmtree(self.tmppath)

View file

@ -16,6 +16,7 @@ def setUp(self):
self.keychain = '/tmp/_test_dedupstore.keychain'
if not os.path.exists(self.keychain):
self.dedupestore('keychain', 'generate')
self.dedupestore('init', self.store_path)
def tearDown(self):
shutil.rmtree(self.tmpdir)