CommonFunction.py 1.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162
  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_video_info(account_id, url):
  14. request_data = {"accountId": account_id}
  15. request = requests.post(url, headers=headers, data=json.JSONEncoder().encode(request_data))
  16. if request['code'] != 0:
  17. return None
  18. return request['data']
  19. def get_video_info(account_id, cnt, url):
  20. request_data = {"accountId": account_id, "videoCnt": cnt}
  21. request = requests.post(url, headers=headers, data=json.JSONEncoder().encode(request_data))
  22. if request['code'] != 0:
  23. return None
  24. return request['data']
  25. def get_video_info(account_id, cnt, url):
  26. request_data = {"accountId": account_id, "videoCnt": cnt}
  27. request = requests.post(url, headers=headers, data=json.JSONEncoder().encode(request_data))
  28. if request['code'] != 0:
  29. return None
  30. return request['data']
  31. def get_video_info(account_id, cnt, start_time, end_time, url):
  32. request_data = {"accountId": account_id, "num": cnt, "startTime": start_time, "endTime": end_time}
  33. request = requests.post(url, headers=headers, data=json.JSONEncoder().encode(request_data))
  34. if request['code'] != 0:
  35. return None
  36. return request['data']
  37. def get_campaign_and_group_name_rule(account_id):
  38. sql = """select campaign_name, unit_name
  39. from ctop_ai_kuaishou_advertiser_strategy
  40. where account_id = %s
  41. and status = 1
  42. limit 1
  43. """ % account_id
  44. df = pd.read_sql(sql, engine)
  45. if df.empty:
  46. return None, None
  47. campaign_name = df['campaign_name'].values[0]
  48. group_name = df['unit_name'].values[0]
  49. return campaign_name, group_name