Merge pull request #9 from quantopian/migration_fixes

Migration fixes

Author: richafrank

bin/pgcontents

@@ -93,7 +93,7 @@ def init(db_url, revision, prompt):
     click.echo("Initializing pgcontents...")
     if prompt:
         click.confirm(CONFIRM_MIGRATION, abort=True)
-        upgrade(db_url, revision)
+    upgrade(db_url, revision)
 
     click.echo('Initialization completed successfully.')
 

pgcontents/checkpoints.py

@@ -60,14 +60,6 @@ def create_file_checkpoint(self, content, format, path):
         with self.engine.begin() as db:
             return save_remote_checkpoint(db, self.user_id, path, b64_content)
 
-    @outside_root_to_404
-    def rename_checkpoint(self, checkpoint_id, old_path, new_path):
-        """Rename a checkpoint from old_path to new_path."""
-        with self.engine.begin() as db:
-            return move_single_remote_checkpoint(
-                db, old_path, new_path, checkpoint_id,
-            )
-
     @outside_root_to_404
     def delete_checkpoint(self, checkpoint_id, path):
         """delete a checkpoint for a file"""

pgcontents/utils/migrate.py

@@ -24,7 +24,7 @@ def temp_alembic_ini(alembic_dir_location, sqlalchemy_url):
         with open(alembic_ini_filename, 'w') as f:
             f.write(
                 ALEMBIC_INI_TEMPLATE.format(
-                    alembic_dir_location=ALEMBIC_DIR_LOCATION,
+                    alembic_dir_location=alembic_dir_location,
                     sqlalchemy_url=sqlalchemy_url,
                 )
             )