Database select enabled only for sql cells

QuantStack · Oct 31, 2023 · fd7279d · fd7279d
1 parent 00f0630
commit fd7279d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -241,7 +241,7 @@ const metadataForm: JupyterFrontEndPlugin<void> = {
     // A widget to associate a database to the cell.
     const databaseSelect: IFormRenderer = {
       fieldRenderer: (props: FieldProps) => {
-        return DatabaseSelect(props);
+        return DatabaseSelect({ ...props, tracker });
       }
     };
     formRegistry.addRenderer(

diff --git a/src/widget.tsx b/src/widget.tsx
@@ -120,7 +120,7 @@ export const SqlSwitchWidget = (options: Private.IOptions): JSX.Element => {
 /**
  * A field including a select to associate a database.
  */
-export const DatabaseSelect = (props: FieldProps) => {
+export const DatabaseSelect = (props: Private.ISelectProps) => {
   const onChange = (event: React.FormEvent) => {
     const value = (event.target as HTMLOptionElement).value;
     const select = props.schema.oneOf?.find(
@@ -131,7 +131,11 @@ export const DatabaseSelect = (props: FieldProps) => {
   return (
     <div>
       <div className="jp-FormGroup-fieldLabel">{props.schema.title}</div>
-      <select onChange={onChange} className={'form-control jp-sqlcell-select'}>
+      <select
+        onChange={onChange}
+        className={'form-control jp-sqlcell-select'}
+        disabled={!SqlCell.isSqlCell(props.tracker.activeCell?.model)}
+      >
         {props.schema.oneOf?.map(oneOf => {
           return <option>{(oneOf as objectEnum).title}</option>;
         })}
@@ -151,4 +155,11 @@ namespace Private {
     commands: CommandRegistry;
     tracker: INotebookTracker;
   }
+
+  /**
+   * The database select props.
+   */
+  export interface ISelectProps extends FieldProps {
+    tracker: INotebookTracker;
+  }
 }