hive.py 1.3 KB

123456789101112131415161718192021222324252627
  1. from app.core.datasource.datasource import DataSourceBase
  2. from app.core.datasource.hive import HiveDS
  3. from configs.settings import DefaultOption, config
  4. from utils import flat_map
  5. HOST = config.get('HIVE', 'HOST')
  6. PORT = config.get('HIVE', 'PORT')
  7. USERNAME = config.get('HIVE', 'USERNAME')
  8. PASSWORD = config.get('HIVE', 'PASSWORD')
  9. DATABASE_NAME = config.get('HIVE', 'DATABASE_NAME')
  10. KERBEROS = config.get('HIVE', 'KERBEROS')
  11. KEYTAB = config.get('HIVE', 'KEYTAB', vars=DefaultOption(config, 'HIVE', KEYTAB = None))
  12. KRB5CONFIG = config.get('HIVE', 'KRB5CONFIG', vars=DefaultOption(config, 'HIVE', KRB5CONFIG = None))
  13. KERBEROS_SERVICE_NAME = config.get('HIVE', 'KERBEROS_SERVICE_NAME', vars=DefaultOption(config, 'HIVE', KERBEROS_SERVICE_NAME = None))
  14. PRINCIPAL = config.get('HIVE', 'PRINCIPAL', vars=DefaultOption(config, 'HIVE', PRINCIPAL = None))
  15. hiveDs = HiveDS(**{'type': 'hive',
  16. 'host': HOST,
  17. 'port': PORT,
  18. 'username': USERNAME,
  19. 'password': PASSWORD,
  20. 'database_name': DATABASE_NAME,
  21. 'kerberos': KERBEROS,
  22. 'keytab': KEYTAB,
  23. 'krb5config': KRB5CONFIG,
  24. 'kerberos_service_name': KERBEROS_SERVICE_NAME,
  25. 'principal':PRINCIPAL})