update mount function to allow location parameter

Author: DenisaCG

jupyter_drives/manager.py

@@ -303,15 +303,17 @@ async def list_drives(self):
         }
         return response
 
-    async def mount_drive(self, drive_name, provider):
+    async def mount_drive(self, drive_name, provider, location=''):
         """Mount a drive.
 
         Args:
             drive_name: name of drive to mount
         """
         try:
             if provider == 's3':
-                if drive_name in self._external_drives and self._external_drives[drive_name]["is_public"] is False: 
+                if location: 
+                    region = location
+                elif drive_name in self._external_drives and self._external_drives[drive_name]["is_public"] is False: 
                     region = self._external_drives[drive_name]["location"]
                 else:
                     region = await self._get_drive_location(drive_name)

src/plugins/drivelistmanager.tsx

@@ -212,7 +212,8 @@ export function DriveListManagerComponent({ model }: IProps) {
   const onAddedPublicDrive = async () => {
     // Check if user has access to drive.
     const result = await mountDrive(publicDrive, {
-      provider: 's3'
+      provider: 's3',
+      location: isPublic ? '' : driveRegion
     });
     if (result && result.error) {
       // Show error in case of failure.

src/requests.ts

@@ -67,11 +67,12 @@ export async function getDrivesList() {
  */
 export async function mountDrive(
   driveName: string,
-  options: { provider: string }
+  options: { provider: string; location?: string }
 ) {
   const body: ReadonlyJSONObject = {
     drive_name: driveName,
-    provider: options.provider
+    provider: options.provider,
+    location: options.location ?? ''
   };
   try {
     await requestAPI<any>('drives', 'POST', body);