import
psycopg2
import
time,sys
class
gpUtil:
def
__init__(
self
,host,user,passwd,db):
self
.host
=
host
self
.user
=
user
self
.passwd
=
passwd
self
.db
=
db
def
db_connect(
self
):
self
.conn
=
psycopg2.connect(host
=
self
.host,user
=
self
.user,password
=
self
.passwd,database
=
self
.db)
self
.cursor
=
self
.conn.cursor()
def
db_fetch_all(
self
,sql):
self
.cursor.execute(sql)
try
:
result
=
self
.cursor.fetchall()
except
:
result
=
"Wrong"
return
result
def
db_close(
self
):
self
.cursor.close()
self
.conn.close()
def
get_list(sql):
conn
=
gpUtil(
'xxxx'
,
'xxxxx'
,
'xxxxx'
,
'xxxxxx'
)
conn.db_connect()
host_list
=
conn.db_fetch_all(sql)
return
host_list
conn.db_close()
def
get_item_list():
tble_list
=
[]
d
=
0
sql
=
re
=
get_list(sql)
for
line
in
re:
line
=
list
(line)
table_name
=
line[
0
]
table_id
=
line[
1
]
table_cd_id
=
line[
2
]
part_name
=
line[
3
]
part_id
=
line[
4
]
part_cd_id
=
line[
5
]
part_sd_id
=
line[
6
]
tble_list.append([table_name,table_id,table_cd_id,part_name,part_id,part_cd_id,part_sd_id])
return
tble_list
def
change_partition(tble_list):
if
len
(tble_list)
=
=
0
:
print
"no need to update"
sys.exit(
1
)
now
=
time.time()
change_content
=
''
rollback_content
=
''
change_sql
=
open
(
"/home/hdfs/bin/hadoop_tools/sql/meta_change"
+
str
(
int
(now))
+
".sql"
,
'w+'
)
rollback_sql
=
open
(
"/home/hdfs/bin/hadoop_tools/sql/meta_rollback"
+
str
(
int
(now))
+
".sql"
,
'w+'
)
for
table_item
in
tble_list:
table_name
=
table_item[
0
]
table_id
=
table_item[
1
]
table_cd_id
=
table_item[
2
]
part_name
=
table_item[
3
]
part_id
=
table_item[
4
]
part_cd_id
=
table_item[
5
]
part_sd_id
=
table_item[
6
]
change_content
+
=
%
(table_name,table_id,table_cd_id,part_name,part_id,part_cd_id,part_sd_id)
rollback_content
+
=
%
(table_name,table_id,table_cd_id,part_name,part_id,part_cd_id,part_sd_id)
change_content
+
=
%
(table_cd_id,part_sd_id)
rollback_content
+
=
%
(part_cd_id,part_sd_id)
change_sql.write(change_content)
rollback_sql.write(rollback_content)
change_sql.close()
rollback_sql.close()
if
__name__
=
=
'__main__'
:
start_time
=
time.time()
tble_list
=
[]
tble_list
=
get_item_list()
change_partition(tble_list)
stop_time
=
time.time()
elapse_time
=
stop_time
-
start_time
print
"elapse_time is %s"
%
(
str
(elapse_time))