Coverage for manila/db/migration.py: 100%

1# Copyright 2010 United States Government as represented by the 

2# Administrator of the National Aeronautics and Space Administration. 

3# All Rights Reserved. 

4# 

5# Licensed under the Apache License, Version 2.0 (the "License"); you may 

6# not use this file except in compliance with the License. You may obtain 

7# a copy of the License at 

8# 

9# http://www.apache.org/licenses/LICENSE-2.0 

10# 

11# Unless required by applicable law or agreed to in writing, software 

12# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT 

13# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the 

14# License for the specific language governing permissions and limitations 

15# under the License. 

16 

17"""Database setup and migration commands.""" 

18 

19from manila import utils 

20 

21 

22IMPL = utils.LazyPluggable( 

23 'db_backend', sqlalchemy='manila.db.migrations.alembic.migration') 

24 

25 

26def upgrade(version): 

27 """Upgrade database to 'version' or the most recent version.""" 

28 return IMPL.upgrade(version) 

29 

30 

31def downgrade(version): 

32 """Downgrade database to 'version' or to initial state.""" 

33 return IMPL.downgrade(version) 

34 

35 

36def version(): 

37 """Display the current database version.""" 

38 return IMPL.version() 

39 

40 

41def stamp(version): 

42 """Stamp database with 'version' or the most recent version.""" 

43 return IMPL.stamp(version) 

44 

45 

46def revision(message, autogenerate): 

47 """Generate new migration script.""" 

48 return IMPL.revision(message, autogenerate)