soltanianalytics opened a new issue #13597:
URL: https://github.com/apache/airflow/issues/13597


   **Apache Airflow version**: 2.0.0
   
   **Environment**: Docker image `apache/airflow:2.0.0-python3.8` on Win10 with 
WSL
   
   **What happened**:
   
   Ich built a custom provider with a number of custom connections.
   
   This works:
   - The connections are properly registered
   - The UI does not show hidden fields as per `get_ui_field_behaviour`
   - The UI correctly relabels fields as per `get_ui_field_behaviour`
   - The UI correctly shows added widgets as per `get_connection_form_widgets` 
(well, mostly)
   
   What does not work:
   - The UI does not save values entered for additional widgets
   
   I used the [JDBC 
example](https://github.com/apache/airflow/blob/master/airflow/providers/jdbc/hooks/jdbc.py)
 to string myself along by copying it and pasting it as a hook into my custom 
provider package. (I did not install the JDBC provider package, unless it is 
installed in the image I use - but if I don't add it in my own provider 
package, I don't have the connection type in the UI, so I assume it is not). 
Curiously, The JDBC hook works just fine. I then created the following file:
   
   ```Python
   """
   You find two child classes of DbApiHook in here. One is the exact copy of 
the JDBC provider hook, minus some irrelevant logc (I only care about the UI 
stuff here). The other is the exact same thing, except I added an "x" behind 
every occurance of "jdbc" in strings and names.
   """
   
   
   from typing import Any, Dict, Optional
   
   from airflow.hooks.dbapi import DbApiHook
   # from airflow.models.connection import Connection
   
   
   class JdbcXHook(DbApiHook):
       """
       Copy of JdbcHook below. Added an "x" at various places, including the 
class name.
       """
   
       conn_name_attr = 'jdbcx_conn_id'  # added x
       default_conn_name = 'jdbcx_default' # added x
       conn_type = 'jdbcx' # added x
       hook_name = 'JDBCx Connection' # added x
       supports_autocommit = True
   
       @staticmethod
       def get_connection_form_widgets() -> Dict[str, Any]:
           """Returns connection widgets to add to connection form"""
           from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
           from flask_babel import lazy_gettext
           from wtforms import StringField
   
           # added an x in the keys
           return {
               "extra__jdbcx__drv_path": StringField(lazy_gettext('Driver 
Path'), widget=BS3TextFieldWidget()),
               "extra__jdbcx__drv_clsname": StringField(
                   lazy_gettext('Driver Class'), widget=BS3TextFieldWidget()
               ),
           }
   
       @staticmethod
       def get_ui_field_behaviour() -> Dict:
           """Returns custom field behaviour"""
           return {
               "hidden_fields": ['port', 'schema', 'extra'],
               "relabeling": {'host': 'Connection URL'},
           }
   
   class JdbcHook(DbApiHook):
       """
       General hook for jdbc db access.
       JDBC URL, username and password will be taken from the predefined 
connection.
       Note that the whole JDBC URL must be specified in the "host" field in 
the DB.
       Raises an airflow error if the given connection id doesn't exist.
       """
   
       conn_name_attr = 'jdbc_conn_id'
       default_conn_name = 'jdbc_default'
       conn_type = 'jdbc'
       hook_name = 'JDBC Connection plain'
       supports_autocommit = True
   
       @staticmethod
       def get_connection_form_widgets() -> Dict[str, Any]:
           """Returns connection widgets to add to connection form"""
           from flask_appbuilder.fieldwidgets import BS3TextFieldWidget
           from flask_babel import lazy_gettext
           from wtforms import StringField
   
           return {
               "extra__jdbc__drv_path": StringField(lazy_gettext('Driver 
Path'), widget=BS3TextFieldWidget()),
               "extra__jdbc__drv_clsname": StringField(
                   lazy_gettext('Driver Class'), widget=BS3TextFieldWidget()
               ),
           }
   
       @staticmethod
       def get_ui_field_behaviour() -> Dict:
           """Returns custom field behaviour"""
           return {
               "hidden_fields": ['port', 'schema', 'extra'],
               "relabeling": {'host': 'Connection URL'},
           }
   ```
   
   **What you expected to happen**:
   
   After doing the above, I expected
   - Seeing both in the add connection UI
   - Being able to use both the same way
   
   What actually happenes
   - I _do_ see both in the UI (Screenshot 1)
   - For some reason, the "normal" hook has BOTH extra fields - not just his 
own two? (Screenshot 2)
   - If I add the connection as in Screenshot 2, they are saved in the four 
fields (his own two + the two for the "x" hook) properly as shown in Screenshot 
3
   - If I seek to edit the connection again, they are also they - all four 
fields - with the correct values in the UI
   - If I add the connection for the "x" type as in Screenshot 4, it ostensibly 
saves it - with two fields as defined in the code
   - You can see in screenshot 5, that the extra is saved as an empty string?!
   - When trying to edit the connection in the UI, you also see that there is 
no data saved for two extra widgets?!
   - I added a few more screenshots of airflow providers CLI command results 
(note that the package `ewah` has a number of other custom hooks, and the issue 
above occurs for *all* of them)
   
   *Screenshot 1:*
   
![image](https://user-images.githubusercontent.com/46958547/104121824-9acc6c00-5341-11eb-821c-4bff40a0e7c7.png)
   
   *Screenshot 2:*
   
![image](https://user-images.githubusercontent.com/46958547/104121854-c94a4700-5341-11eb-8d3c-80b6380730d9.png)
   
   *Screenshot 3:*
   
![image](https://user-images.githubusercontent.com/46958547/104121912-247c3980-5342-11eb-8030-11c7348309f3.png)
   
   *Screenshot 4:*
   
![image](https://user-images.githubusercontent.com/46958547/104121944-5e4d4000-5342-11eb-83b7-870711ccd367.png)
   
   *Screenshot 5:*
   
![image](https://user-images.githubusercontent.com/46958547/104121971-82a91c80-5342-11eb-83b8-fee9386c0c4f.png)
   
   *Screenshot 6 - airflow providers behaviours:*
   
![image](https://user-images.githubusercontent.com/46958547/104122073-1c70c980-5343-11eb-88f6-6130e5de9e92.png)
   
   *Screenshot 7 - airflow providers get:*
   
![image](https://user-images.githubusercontent.com/46958547/104122092-41fdd300-5343-11eb-9bda-f6849812ba56.png)
   (Note: This error occurs with pre-installed providers as well)
   
   *Screenshot 8 - airflow providers hooks:*
   
![image](https://user-images.githubusercontent.com/46958547/104122109-65288280-5343-11eb-8322-dda73fef6649.png)
   
   *Screenshot 9 - aorflow providers list:*
   
![image](https://user-images.githubusercontent.com/46958547/104122191-c94b4680-5343-11eb-80cf-7f510d4b6e9a.png)
   
   
   *Screenshot 10 - airflow providers widgets:*
   
![image](https://user-images.githubusercontent.com/46958547/104122142-930dc700-5343-11eb-96be-dec43d87a59d.png)
   
   
   **How to reproduce it**:
   
   - create a custom provider package
   - add the code snippet pasted above somewhere
   - add the two classes to the `hook-class-names` list in the provider info
   - install the provider package
   - do what I described above
   
   


----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
[email protected]


Reply via email to