修改统计指标

eda/gray_stat/ctr.py

@@ -4,23 +4,6 @@ import time
 import sys
 
 
-OUTPUT_PATH = "/data2/models/eda/gray_stat/"
-
-num = sys.argv[1]
-if num == "3|4":
-    OUTPUT_PATH = "/data2/models/eda/gray_stat/3|4"
-    table_name = "jerry_prod.nd_device_cid_similarity_matrix_tmp"
-elif num == "5|6":
-    OUTPUT_PATH = "/data2/models/eda/gray_stat/5|6"
-    table_name = "eagle.ffm_diary_queue_temp"
-elif num == "7|8":
-    OUTPUT_PATH = "/data2/models/eda/gray_stat/7|8"
-    table_name = "eagle.search_queue"
-else:
-    print("参数必须是 '3|4' 或 '5|6' 或 '7|8'")
-    print("程序异常停止，请重新输入参数运行")
-    sys.exit()
-
 class GrayStat(object):
 	def __init__(self, cid_type, uid_type, ndays=get_yesterday_date()):
 		"""
@@ -125,8 +108,8 @@ def main():
 	output = OUTPUT_PATH + "ctr.csv"
 	with open(output,"a+") as f:
 		print("stat" + "  " + get_yesterday_date())
-		g_class = GrayStat("diary","3|4")
-		a_class = AllStat("diary","3|4")
+		g_class = GrayStat("diary",num)
+		a_class = AllStat("diary",num)
 		line1 = str(g_class.get_uid_count())+","+str(g_class.get_uid_imp_times())+","+str(g_class.get_uid_clk_times())
 		line2 = str(a_class.get_uid_count())+","+str(a_class.get_uid_imp_times())+","+str(a_class.get_uid_clk_times())
 		g_ctr = g_class.get_uid_clk_times()/g_class.get_uid_imp_times()
@@ -139,4 +122,19 @@ def main():
 
 
 if __name__ == '__main__':
+	OUTPUT_PATH = "/data2/models/eda/gray_stat/"
+	num = sys.argv[1]
+	if num == "3|4":
+		OUTPUT_PATH = "/data2/models/eda/gray_stat/3|4"
+		table_name = "jerry_prod.nd_device_cid_similarity_matrix_tmp"
+	elif num == "5|6":
+		OUTPUT_PATH = "/data2/models/eda/gray_stat/5|6"
+		table_name = "eagle.ffm_diary_queue_temp"
+	elif num == "7|8":
+		OUTPUT_PATH = "/data2/models/eda/gray_stat/7|8"
+		table_name = "eagle.search_queue"
+	else:
+		print("参数必须是 '3|4' 或 '5|6' 或 '7|8'")
+		print("程序异常停止，请重新输入参数运行")
+		sys.exit()
 	main()
\ No newline at end of file