本文研究的主要问题是python语言导出hive数据表的schema,分享了实现代码,具体如下。
为了避免运营提出无穷无尽的查询需求,我们决定将有查询价值的数据从mysql导入hive中,让他们使用HUE这个开源工具进行查询。想必他们对表结构不甚了解,还需要为之提供一个表结构说明,于是编写了一个脚本,从hive数据库中将每张表的字段即类型查询出来,代码如下:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
#coding=utf-8
import pyhs2
from xlwt import *
hiveconn = pyhs2.connect(host = '10.46.77.120' ,
port = 10000 ,
authMechanism = 'PLAIN' ,
user = 'hadoop' ,
database = 'hibiscus_data' ,
)
def create_excel():
sql = 'show tables'
tables = []
with hiveconn.cursor() as cursor:
cursor.execute(sql)
res = cursor.fetch()
for table in res:
tables.append(table[ 0 ])
tableinfo = []
for table in tables:
tableinfo.append(get_column_info(table))
create_excel_ex(tableinfo)
def create_excel_ex(tableinfo):
w = Workbook()
sheet = w.add_sheet(u '表结构' )
row = 0
for info in tableinfo:
row = write_tale_info(info,sheet,row)
w.save( 'hive_schema.xls' )
def write_tale_info(tableinfo,sheet,row):
print row
sheet.write_merge(row,row, 0 , 2 ,tableinfo[ 'table' ])
row + = 1
sheet.write(row, 0 ,u '名称' )
sheet.write(row, 1 ,u '类型' )
sheet.write(row, 2 ,u '解释' )
row + = 1
fields = tableinfo[ 'fields' ]
for field in fields:
sheet.write(row, 0 ,field[ 'name' ])
sheet.write(row, 1 ,field[ 'type' ])
row + = 1
return row + 1
def get_column_info(table):
sql = 'desc {table}' . format (table = table)
info = { 'table' :table, 'fields' :[]}
with hiveconn.cursor() as cursor:
cursor.execute(sql)
res = cursor.fetch()
for item in res:
if item[ 0 ] = = '':
break
info[ 'fields' ].append({ 'name' :item[ 0 ], 'type' :item[ 1 ]})
return info
if __name__ = = '__main__' :
create_excel()
|
其实,我们的hive数据库将所有的元数据存储在了mysql当中,分析这些元数据也可以获得表结构信息。
总结
以上就是本文关于python导出hive数据表的schema实例代码的全部内容,希望对大家有所帮助。感兴趣的朋友可以继续参阅本站其他相关专题,如有不足之处,欢迎留言指出。感谢朋友们对本站的支持!
原文链接:http://blog.csdn.net/kwsy2008/article/details/52041811