CommonFunction.py 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. import requests
  2. from utils.UrlConfig import *
  3. import json
  4. import pandas as pd
  5. from utils.DataBaseConfig import *
  6. def get_lower_case_name(name):
  7. lst = []
  8. for index, char in enumerate(name):
  9. if char.isupper() and index != 0:
  10. lst.append("_")
  11. lst.append(char)
  12. return "".join(lst).lower()
  13. def get_new_video_info(account_id, url):
  14. request_data = {"accountId": account_id}
  15. request = requests.post(url, headers=headers, data=json.JSONEncoder().encode(request_data)).text
  16. request = json.loads(request)
  17. if request['code'] != 0:
  18. return None
  19. return request['data']
  20. def get_missing_video_info(account_id, cnt, url, start_time, end_time):
  21. request_data = {"accountId": account_id, "videoCnt": cnt, "startTime": start_time, "endTime": end_time}
  22. request = requests.post(url, headers=headers, data=json.JSONEncoder().encode(request_data)).text
  23. request = json.loads(request)
  24. if request['code'] != 0:
  25. return None
  26. return request['data']
  27. def get_top_video_info(account_id, cnt, start_time, end_time, url):
  28. request_data = {"accountId": account_id, "num": cnt, "startTime": start_time, "endTime": end_time}
  29. request = requests.post(url, headers=headers, data=json.JSONEncoder().encode(request_data)).text
  30. request = json.loads(request)
  31. if request['code'] != 0:
  32. return None
  33. return request['data']
  34. def get_campaign_and_group_name_rule(account_id):
  35. sql = """select campaign_name, group_name
  36. from ctop_ai_kuaishou_advertiser_strategy
  37. where account_id = %s
  38. and status = 1
  39. limit 1
  40. """ % account_id
  41. df = pd.read_sql(sql, engine)
  42. if df.empty:
  43. return None, None
  44. campaign_name = df['campaign_name'].values[0]
  45. group_name = df['group_name'].values[0]
  46. return campaign_name, group_name