快捷導(dǎo)航

Python實(shí)現(xiàn)Kerberos用戶的增刪改查操作

更新時(shí)間：2020年12月14日 09:07:52 作者：源神

這篇文章主要介紹了Python實(shí)現(xiàn)Kerberos用戶的增刪改查操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

1、首先模擬python類似shell命令行操作的接口：

python安裝subprocess（本地）、paramiko（SSH遠(yuǎn)程）

#-*- coding: UTF-8 -*-
#!/usr/bin/python
import os, sys
import subprocess
import paramiko
import settings
 
class RunCmd(object):
 def __init__(self):
  self.cmd = 'ls'
 
 @staticmethod
 def local_run(cmd):
  print('start executing...')
  print('cmd is -------> %s' % str(cmd))
  s = subprocess.Popen(str(cmd), shell=True, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
  out, err = s.communicate()
  print("outinfo is -------> %s" % out)
  print("errinfo is -------> %s" % err)
  print('finish executing...')
  print('result:------> %s' % s.returncode)
  return s.returncode
 
 @staticmethod
 def remote_run(host, username, password, port, cmd):
  client = paramiko.SSHClient()
  client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
  client.connect(hostname=host, port=int(port), username=username, password=password, timeout=5)
  stdin, stdout, stderr = client.exec_command(cmd)
  result = stdout.read()
  client.close()
  return result
 
 @staticmethod
 def krb_run(cmd):
  print('krb_run start...')
  print('cmd is -------> %s' % str(cmd))
  result = RunCmd.remote_run(settings.KRB_HOST, settings.USERNAME, settings.PASSWORD, settings.PORT, cmd)
  print('result:------> %s' % result)
  print('krb_run finish...')
  return result

2、Kerberos常用的命令操作封裝成接口，其他簡(jiǎn)單。但需要交互的是刪除 principal

 def delete_user(self, username):
  cmd = r"""
   expect -c "
   set timeout 1;
   spawn kadmin.local -q \"delete_principal {principal}\" ;
   expect yes/no {{ send \"yes\r\" }} ;
   expect *\r
   expect \r
   expect eof
   "
  """.format(principal=username)
  RunCmd.krb_run(cmd)

補(bǔ)充知識(shí)：python操作有Kerberos認(rèn)證的hive庫(kù)

之前訪問(wèn)hive都比較簡(jiǎn)單，直接用pyhive連接即可。

但是最近遇到了一個(gè)問(wèn)題，hive有了Kerberosren認(rèn)證。

最終經(jīng)過(guò)各種嘗試和靈感迸發(fā)，終于解決了這個(gè)問(wèn)題，遂記錄之。

代碼

from pyhive.hive import connect
con = connect(host='XXXX',port=10000,auth='KERBEROS',kerberos_service_name="hive")
cursor = con.cursor()
cursor.execute('select * from tmp.pricing_calculate_result_spark where time_id="201907171355" limit 10,1')
datas = cursor.fetchall()
print(datas)
cursor.close()
con.close()

端口和ip都換成自己的，auth和kerberos_service_name不要改

運(yùn)行效果