profile_cleaner.py 910 B

1234567891011121314151617181920212223
  1. #! /usr/bin/env python
  2. # -*- coding: utf-8 -*-
  3. # vim:fenc=utf-8
  4. """
  5. 用户画像信息提取在不断迭代中,可能需清除旧的用户画像信息避免干扰,故提供此工具
  6. """
  7. from pqai_agent import configs
  8. from pqai_agent.user_manager import MySQLUserManager
  9. if __name__ == '__main__':
  10. config = configs.get()
  11. user_db_config = config['storage']['user']
  12. staff_db_config = config['storage']['staff']
  13. user_manager = MySQLUserManager(user_db_config['mysql'], user_db_config['table'], staff_db_config['table'])
  14. user_ids_to_clean = ['7881299986081786', '7881303544096524', '7881300732152777', '7881301752098239', '7881299457990953', '7881302872936170',]
  15. for user_id in user_ids_to_clean:
  16. profile = user_manager.get_user_profile(user_id)
  17. profile['name'] = ''
  18. profile['preferred_nickname'] = ''
  19. user_manager.save_user_profile(user_id, profile)