update backend to add support for external drives

Author: DenisaCG

jupyter_drives/handlers.py

@@ -101,8 +101,8 @@ async def post(self, drive: str = "", path: str = ""):
         body = self.get_json_body()
         if 'location' in body:
             result = await self._manager.new_drive(drive, **body)
-        if 'public' in body:
-            result = await self._manager.add_public_drive(drive)
+        if 'is_public' in body:
+            result = await self._manager.add_external_drive(drive, **body)
         else:
             result = await self._manager.new_file(drive, path, **body)
         self.finish(result)

jupyter_drives/manager.py

@@ -311,7 +311,11 @@ async def mount_drive(self, drive_name, provider):
         """
         try:
             if provider == 's3':
-                region = await self._get_drive_location(drive_name)
+                if drive_name in self._external_drives and self._external_drives[drive_name]["is_public"] is False: 
+                    print('FOUND DRIVE IN EXTERNAL TO MOUNT: ', self._external_drives[drive_name]["location"])
+                    region = self._external_drives[drive_name]["location"]
+                else:
+                    region = await self._get_drive_location(drive_name)
             self._initialize_content_manager(drive_name, provider, region)
         except Exception as e:
             raise tornado.web.HTTPError(
@@ -712,18 +716,21 @@ async def new_drive(self, new_drive_name, location='us-east-1'):
 
         return
 
-    async def add_public_drive(self, drive_name):
+    async def add_external_drive(self, drive_name, is_public, region='us-east-1'):
         """Mount a drive.
 
         Args:
             drive_name: name of public bucket to mount
         """
         try:
             drive = {
-                "is_public": True,
-                "url": drive_name
+                "is_public": is_public,
+                "url": drive_name,
+                "location": region
             };
+            print('ADDED DRIVE: ', drive)
             self._external_drives[drive_name] = drive;
+            print('NEW EXTERNAL DRIVEs: ', self._external_drives)
         except Exception as e:
             raise tornado.web.HTTPError(
             status_code= httpx.codes.BAD_REQUEST,