mindful.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "b6770a25",
   "metadata": {},
   "outputs": [],
   "source": [
    "import psutil\n",
    "import pywinauto\n",
    "from pywinauto.application import Application\n",
    "import sys\n",
    "import pickle"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "d1131dc6",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Collecting pywinauto\n",
      "  Using cached pywinauto-0.6.8-py2.py3-none-any.whl (362 kB)\n",
      "Requirement already satisfied: six in d:\\programfiles\\conda\\lib\\site-packages (from pywinauto) (1.16.0)\n",
      "Requirement already satisfied: pywin32 in d:\\programfiles\\conda\\lib\\site-packages (from pywinauto) (302)\n",
      "Requirement already satisfied: comtypes in d:\\programfiles\\conda\\lib\\site-packages (from pywinauto) (1.1.10)\n",
      "Installing collected packages: pywinauto\n",
      "Successfully installed pywinauto-0.6.8\n",
      "Note: you may need to restart the kernel to use updated packages.\n"
     ]
    }
   ],
   "source": [
    "%pip install pywinauto"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "10588c5f",
   "metadata": {
    "scrolled": false
   },
   "outputs": [],
   "source": [
    "PID = 0\n",
    "for proc in psutil.process_iter():\n",
    "    try:\n",
    "        pinfo = proc.as_dict(attrs=['pid', 'name'])\n",
    "    except psutil.NoSuchProcess:\n",
    "        pass\n",
    "    else:\n",
    "        if 'WeChat.exe' == pinfo['name']:\n",
    "            PID = pinfo['pid']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 138,
   "id": "c6e76911",
   "metadata": {},
   "outputs": [],
   "source": [
    "app = Application(backend='uia').connect(process=PID)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 153,
   "id": "09f5a813",
   "metadata": {
    "scrolled": true
   },
   "outputs": [],
   "source": [
    "chat_win = app['小狐狸🦊']\n",
    "#print(chat_win.dump_tree())\n",
    "#print(\"-----------------\")\n",
    "#print(chats)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 227,
   "id": "9c63dbde",
   "metadata": {},
   "outputs": [],
   "source": [
    "def search(message):\n",
    "    msg_contents = message.children()[0].children()[1].children()\n",
    "    time = msg_contents[0].children()[2].window_text()\n",
    "    author = msg_contents[0].children()[0].window_text()\n",
    "    try:\n",
    "        msg = msg_contents[1].children()[0].children()[0].children()[0].window_text()\n",
    "    except IndexError as e:\n",
    "        msg = \"[图片]\"\n",
    "    if not msg:\n",
    "        msg = \"[动画表情]\"\n",
    "    return time, author, msg\n",
    "\n",
    "class RawMessage:\n",
    "    def __init__(self, time, author, msg):\n",
    "        self.time = time\n",
    "        self.author = author\n",
    "        self.msg = msg\n",
    "\n",
    "all_msg = []\n",
    "for cycle in range(0, 50):\n",
    "    chats = chat_win.wrapper_object().descendants()\n",
    "    current_round = []\n",
    "    for message in chats:\n",
    "        classname = message.friendly_class_name()\n",
    "        if (classname == \"ListItem\"):\n",
    "            time, author, msg = search(message)\n",
    "            #print(\"---------\")\n",
    "            #print(time)\n",
    "            #print(author)\n",
    "            #print(msg)\n",
    "            current_round.append(RawMessage(time, author, msg))\n",
    "    all_msg.extend(current_round)\n",
    "    cords = chat_win.rectangle()\n",
    "    pywinauto.mouse.scroll(wheel_dist=4, coords=(cords.left+10, cords.bottom-10))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 230,
   "id": "5b28a792",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "昨天 | 小狐狸🦊: traffic不是london吗\n",
      "昨天 | Andy: [Grimace]\n",
      "昨天 | Andy: 除了rio我不知道有啥了\n",
      "昨天 | Andy: 宝宝要不你帮我整理下[Toasted]\n",
      "昨天 | 小狐狸🦊: 1\n",
      "昨天 | Andy: oh是吗\n",
      "昨天 | 小狐狸🦊: 来的*\n",
      "昨天 | 小狐狸🦊: 你的stats哪来来的\n",
      "昨天 | 小狐狸🦊: 我看看吧\n",
      "昨天 | 小狐狸🦊: ok\n",
      "昨天 | 小狐狸🦊: gpt？\n",
      "昨天 | Andy: gpt\n",
      "昨天 | Andy: 数据有问题吗\n",
      "昨天 | Andy: 应该没有吧？\n",
      "昨天 | 小狐狸🦊: 我等会看看\n",
      "昨天 | 小狐狸🦊: 不知道诶\n",
      "昨天 | Andy: ？\n",
      "昨天 | 小狐狸🦊: [图片]\n",
      "昨天 | 小狐狸🦊: btw\n",
      "昨天 | 小狐狸🦊: 乐的\n",
      "昨天 | Andy: 荒谬\n",
      "昨天 | 小狐狸🦊: 哥们儿想冲数学领奖台？？\n",
      "昨天 | Andy: bevis有竞争对手了\n",
      "昨天 | 小狐狸🦊: [Doge][Doge][Doge]\n",
      "昨天 | 小狐狸🦊: karen有竞争对手了\n",
      "昨天 | Andy: [Doge][Doge][Doge]\n",
      "昨天 | Andy: 哈哈哈哈哈哈哈哈哈哈哈哈哈哈\n",
      "昨天 | 小狐狸🦊: 她今天不在学校\n",
      "昨天 | 小狐狸🦊: 下周吧\n",
      "昨天 | Andy: 去找igor了吗\n",
      "昨天 | Andy: karen怎么样\n",
      "昨天 | 小狐狸🦊: 她肯定上\n",
      "昨天 | 小狐狸🦊: 不是我说\n",
      "昨天 | 小狐狸🦊: 我赌karen肯定上\n",
      "昨天 | Andy: Kinda like ur geo\n",
      "昨天 | 小狐狸🦊: 真的\n",
      "昨天 | Andy: 不上就离谱\n",
      "昨天 | Andy: 她全满的\n",
      "昨天 | Andy: [动画表情]\n",
      "昨天 | 小狐狸🦊: [动画表情]\n",
      "8:02 am | Andy: 我现在去找\n",
      "8:02 am | 小狐狸🦊: ？\n",
      "8:02 am | Andy: 收到\n",
      "8:02 am | 小狐狸🦊: [图片]\n",
      "8:02 am | Andy: ？\n",
      "8:02 am | 小狐狸🦊: 不用找了\n",
      "8:02 am | 小狐狸🦊: 我不会\n",
      "8:02 am | Andy: 阴间\n",
      "8:02 am | Andy: 66\n",
      "8:02 am | 小狐狸🦊: 7m是cbd\n",
      "8:03 am | 小狐狸🦊: 死\n",
      "8:03 am | 小狐狸🦊: 怎么办\n",
      "8:03 am | Andy: [动画表情]\n",
      "8:03 am | 小狐狸🦊: 收到\n",
      "8:03 am | Andy: 心底的\n",
      "8:03 am | Andy: [图片]\n",
      "8:03 am | Andy: [Toasted]\n",
      "8:03 am | Andy: 。\n",
      "8:03 am | 小狐狸🦊: 但是不一定考那个\n",
      "8:04 am | Andy: 那咋办\n",
      "8:04 am | Andy: 急\n",
      "8:04 am | Andy: 。。[Toasted][Toasted][Toasted][Toasted][Toasted][Toasted]\n",
      "8:03 am | 小狐狸🦊: 不然也不会放在桌子上吧\n",
      "8:05 am | 小狐狸🦊: [动画表情]\n",
      "8:13 am | 小狐狸🦊: i found ittt\n",
      "8:14 am | 小狐狸🦊: 他们昨天考的是这一套 7m换了\n",
      "8:13 am | 小狐狸🦊: 但是没什么用\n",
      "8:13 am | 小狐狸🦊: [图片]\n",
      "8:15 am | Andy: 你下节课请假吗\n",
      "8:14 am | Andy: tyty\n",
      "8:14 am | Andy: damn\n",
      "8:15 am | 小狐狸🦊: i wish\n",
      "8:16 am | Andy: 现在叫你请还来得及吗\n",
      "8:16 am | 小狐狸🦊: 是谁不叫我请的\n",
      "8:15 am | 小狐狸🦊: 但是没有\n",
      "8:16 am | Andy: 我想清\n",
      "8:16 am | Andy: 就请一节\n",
      "8:16 am | 小狐狸🦊: 你怎么请\n",
      "8:16 am | 小狐狸🦊: 、？\n",
      "8:16 am | 小狐狸🦊: 怎么\n",
      "8:17 am | Andy: no\n",
      "8:17 am | 小狐狸🦊: 校医？\n",
      "8:16 am | Andy: sickleave\n",
      "8:16 am | Andy: ？\n",
      "8:17 am | Andy: 上午不用\n",
      "8:17 am | 小狐狸🦊: how then\n",
      "8:17 am | 小狐狸🦊: cms吗\n",
      "8:18 am | Andy: 我还没请\n",
      "8:17 am | 小狐狸🦊: 好讨厌你呀\n",
      "8:17 am | 小狐狸🦊: brooo\n",
      "8:17 am | Andy: yes\n",
      "8:19 am | 小狐狸🦊: 正好 我语文没复习\n",
      "8:19 am | 小狐狸🦊: 我试试\n",
      "8:19 am | 小狐狸🦊: 行\n",
      "8:18 am | Andy: （先劝你一起请\n",
      "8:28 am | Andy: [图片]\n",
      "8:19 am | Andy: [Grimace]\n",
      "8:19 am | Andy: 我也这么想的\n",
      "8:29 am | 小狐狸🦊: 厉害\n",
      "8:29 am | 小狐狸🦊: [图片]\n",
      "8:41 am | 小狐狸🦊: 等会来找我\n",
      "8:41 am | 小狐狸🦊: 我在510\n",
      "8:29 am | Andy: lol\n",
      "8:45 am | Andy: 好嘞\n",
      "12:38 pm | 小狐狸🦊: [图片]\n",
      "9:21 am | Andy: 510？\n",
      "1:05 pm | Andy: [图片]\n",
      "4:48 pm | 小狐狸🦊: 好 明天见\n",
      "4:46 pm | Andy: k\n",
      "4:42 pm | 小狐狸🦊: 算了 还是下次再给你吧 等你不生气的时候TwT\n",
      "5:06 pm | Andy: 好吧\n",
      "5:06 pm | Andy: 只是觉得第一个520\n",
      "5:06 pm | Andy: 前面有点生气\n",
      "5:06 pm | Andy: 对不起喽\n",
      "5:09 pm | Andy: 好嘞\n",
      "5:08 pm | 小狐狸🦊: 没事啦 我只是有点不理解 你也知道我很想和你出去玩 只是还有九天就考试了 考完我们好好去玩okkk?\n",
      "5:07 pm | Andy: [动画表情]\n",
      "5:11 pm | Andy: 但廿一宝宝要求可能不一样\n",
      "5:10 pm | 小狐狸🦊: 你是比较的厉害[CoolGuy]\n",
      "5:09 pm | Andy: 我没把分看的那么重[Slight]随便考考也有A*\n",
      "5:11 pm | Andy: nah\n",
      "5:11 pm | 小狐狸🦊: 我虽然可能maybe考的还行 但是我理解东西所需要的时间比其他人多很多\n",
      "5:11 pm | Andy: [动画表情]\n",
      "5:11 pm | Andy: 我要90\n",
      "5:11 pm | 小狐狸🦊: in order words我笨笨的[CoolGuy]\n",
      "5:11 pm | Andy: 你要100\n",
      "5:11 pm | Andy: thats the diff\n",
      "5:12 pm | 小狐狸🦊: [动画表情]\n",
      "5:12 pm | Andy: 不打扰喽\n",
      "5:12 pm | Andy: 你去内卷吧\n",
      "9:30 pm | Andy: 明天记得多和我拍点照\n",
      "9:29 pm | Andy: 好嘞！\n",
      "9:29 pm | 小狐狸🦊: [Kiss][Rose]\n",
      "9:29 pm | 小狐狸🦊: 明天记得穿polo！！\n",
      "9:30 pm | Andy: （既然某人过于内卷不愿意出去玩\n",
      "9:30 pm | 小狐狸🦊: 好！！\n",
      "9:30 pm | Andy: 那个礼物是啥啊\n",
      "9:30 pm | Andy: 哦对了\n",
      "9:31 pm | Andy: but tell me tmr\n",
      "9:31 pm | Andy: 给点hint\n",
      "9:30 pm | 小狐狸🦊: 真的想知道吗！！\n",
      "9:30 pm | Andy: [动画表情]\n",
      "9:32 pm | Andy: 靠\n",
      "9:31 pm | 小狐狸🦊: [Drool]\n",
      "9:31 pm | 小狐狸🦊: or green\n",
      "9:31 pm | 小狐狸🦊: something not blue\n",
      "9:32 pm | Andy: 我没听\n",
      "9:32 pm | Andy: 明天几点到来着\n",
      "9:32 pm | Andy: miles说\n",
      "9:33 pm | Andy: o\n",
      "9:33 pm | 小狐狸🦊: [图片]\n",
      "9:33 pm | Andy: [动画表情]\n",
      "9:33 pm | 小狐狸🦊: basketballcourt\n",
      "9:33 pm | 小狐狸🦊: 7:45吧\n"
     ]
    }
   ],
   "source": [
    "def checkSame(a, b):\n",
    "    if a.time == b.time and a.author == b.author and a.msg == b.msg:\n",
    "        return True\n",
    "    return False\n",
    "\n",
    "def checkSeen(l, a):\n",
    "    for i in l:\n",
    "        if checkSame(i, a):\n",
    "            return True\n",
    "    return False\n",
    "\n",
    "def printRawMsg(message):\n",
    "    print(f\"{message.time} | {message.author}: {message.msg}\")\n",
    "\n",
    "# Will result in less messages than actual. Fine with me though, I don't know how to process it properly.\n",
    "seen = set()\n",
    "uniq = []\n",
    "for msg in all_msg:\n",
    "    if not checkSeen(seen, msg):\n",
    "        uniq.append(msg)\n",
    "        seen.add(msg)\n",
    "\n",
    "# all_msg = uniq[::-1]\n",
    "all_msg = uniq\n",
    "for msg in all_msg:\n",
    "    printRawMsg(msg)\n",
    "    continue\n",
    "    \n",
    "# TODO: The order of the messages if fucked up for some reason. I guess we just have to ignore message order within the minute..."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 285,
   "id": "db88a886",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "2023-05-17 NA | 小狐狸🦊: traffic不是london吗\n",
      "2023-05-17 NA | Andy: [Grimace]\n",
      "2023-05-17 NA | Andy: 除了rio我不知道有啥了\n",
      "2023-05-17 NA | Andy: 宝宝要不你帮我整理下[Toasted]\n",
      "2023-05-17 NA | 小狐狸🦊: 1\n",
      "2023-05-17 NA | Andy: oh是吗\n",
      "2023-05-17 NA | 小狐狸🦊: 来的*\n",
      "2023-05-17 NA | 小狐狸🦊: 你的stats哪来来的\n",
      "2023-05-17 NA | 小狐狸🦊: 我看看吧\n",
      "2023-05-17 NA | 小狐狸🦊: ok\n",
      "2023-05-17 NA | 小狐狸🦊: gpt？\n",
      "2023-05-17 NA | Andy: gpt\n",
      "2023-05-17 NA | Andy: 数据有问题吗\n",
      "2023-05-17 NA | Andy: 应该没有吧？\n",
      "2023-05-17 NA | 小狐狸🦊: 我等会看看\n",
      "2023-05-17 NA | 小狐狸🦊: 不知道诶\n",
      "2023-05-17 NA | Andy: ？\n",
      "2023-05-17 NA | 小狐狸🦊: [图片]\n",
      "2023-05-17 NA | 小狐狸🦊: btw\n",
      "2023-05-17 NA | 小狐狸🦊: 乐的\n",
      "2023-05-17 NA | Andy: 荒谬\n",
      "2023-05-17 NA | 小狐狸🦊: 哥们儿想冲数学领奖台？？\n",
      "2023-05-17 NA | Andy: bevis有竞争对手了\n",
      "2023-05-17 NA | 小狐狸🦊: [Doge][Doge][Doge]\n",
      "2023-05-17 NA | 小狐狸🦊: karen有竞争对手了\n",
      "2023-05-17 NA | Andy: [Doge][Doge][Doge]\n",
      "2023-05-17 NA | Andy: 哈哈哈哈哈哈哈哈哈哈哈哈哈哈\n",
      "2023-05-17 NA | 小狐狸🦊: 她今天不在学校\n",
      "2023-05-17 NA | 小狐狸🦊: 下周吧\n",
      "2023-05-17 NA | Andy: 去找igor了吗\n",
      "2023-05-17 NA | Andy: karen怎么样\n",
      "2023-05-17 NA | 小狐狸🦊: 她肯定上\n",
      "2023-05-17 NA | 小狐狸🦊: 不是我说\n",
      "2023-05-17 NA | 小狐狸🦊: 我赌karen肯定上\n",
      "2023-05-17 NA | Andy: Kinda like ur geo\n",
      "2023-05-17 NA | 小狐狸🦊: 真的\n",
      "2023-05-17 NA | Andy: 不上就离谱\n",
      "2023-05-17 NA | Andy: 她全满的\n",
      "2023-05-17 NA | Andy: [动画表情]\n",
      "2023-05-17 NA | 小狐狸🦊: [动画表情]\n",
      "2023-05-18 08:02:00 | Andy: 我现在去找\n",
      "2023-05-18 08:02:00 | 小狐狸🦊: ？\n",
      "2023-05-18 08:02:00 | Andy: 收到\n",
      "2023-05-18 08:02:00 | 小狐狸🦊: [图片]\n",
      "2023-05-18 08:02:00 | Andy: ？\n",
      "2023-05-18 08:02:00 | 小狐狸🦊: 不用找了\n",
      "2023-05-18 08:02:00 | 小狐狸🦊: 我不会\n",
      "2023-05-18 08:02:00 | Andy: 阴间\n",
      "2023-05-18 08:02:00 | Andy: 66\n",
      "2023-05-18 08:02:00 | 小狐狸🦊: 7m是cbd\n",
      "2023-05-18 08:03:00 | 小狐狸🦊: 死\n",
      "2023-05-18 08:03:00 | 小狐狸🦊: 怎么办\n",
      "2023-05-18 08:03:00 | Andy: [动画表情]\n",
      "2023-05-18 08:03:00 | 小狐狸🦊: 收到\n",
      "2023-05-18 08:03:00 | Andy: 心底的\n",
      "2023-05-18 08:03:00 | Andy: [图片]\n",
      "2023-05-18 08:03:00 | Andy: [Toasted]\n",
      "2023-05-18 08:03:00 | Andy: 。\n",
      "2023-05-18 08:03:00 | 小狐狸🦊: 但是不一定考那个\n",
      "2023-05-18 08:04:00 | Andy: 那咋办\n",
      "2023-05-18 08:04:00 | Andy: 急\n",
      "2023-05-18 08:04:00 | Andy: 。。[Toasted][Toasted][Toasted][Toasted][Toasted][Toasted]\n",
      "2023-05-18 08:03:00 | 小狐狸🦊: 不然也不会放在桌子上吧\n",
      "2023-05-18 08:05:00 | 小狐狸🦊: [动画表情]\n",
      "2023-05-18 08:13:00 | 小狐狸🦊: i found ittt\n",
      "2023-05-18 08:14:00 | 小狐狸🦊: 他们昨天考的是这一套 7m换了\n",
      "2023-05-18 08:13:00 | 小狐狸🦊: 但是没什么用\n",
      "2023-05-18 08:13:00 | 小狐狸🦊: [图片]\n",
      "2023-05-18 08:15:00 | Andy: 你下节课请假吗\n",
      "2023-05-18 08:14:00 | Andy: tyty\n",
      "2023-05-18 08:14:00 | Andy: damn\n",
      "2023-05-18 08:15:00 | 小狐狸🦊: i wish\n",
      "2023-05-18 08:16:00 | Andy: 现在叫你请还来得及吗\n",
      "2023-05-18 08:16:00 | 小狐狸🦊: 是谁不叫我请的\n",
      "2023-05-18 08:15:00 | 小狐狸🦊: 但是没有\n",
      "2023-05-18 08:16:00 | Andy: 我想清\n",
      "2023-05-18 08:16:00 | Andy: 就请一节\n",
      "2023-05-18 08:16:00 | 小狐狸🦊: 你怎么请\n",
      "2023-05-18 08:16:00 | 小狐狸🦊: 、？\n",
      "2023-05-18 08:16:00 | 小狐狸🦊: 怎么\n",
      "2023-05-18 08:17:00 | Andy: no\n",
      "2023-05-18 08:17:00 | 小狐狸🦊: 校医？\n",
      "2023-05-18 08:16:00 | Andy: sickleave\n",
      "2023-05-18 08:16:00 | Andy: ？\n",
      "2023-05-18 08:17:00 | Andy: 上午不用\n",
      "2023-05-18 08:17:00 | 小狐狸🦊: how then\n",
      "2023-05-18 08:17:00 | 小狐狸🦊: cms吗\n",
      "2023-05-18 08:18:00 | Andy: 我还没请\n",
      "2023-05-18 08:17:00 | 小狐狸🦊: 好讨厌你呀\n",
      "2023-05-18 08:17:00 | 小狐狸🦊: brooo\n",
      "2023-05-18 08:17:00 | Andy: yes\n",
      "2023-05-18 08:19:00 | 小狐狸🦊: 正好 我语文没复习\n",
      "2023-05-18 08:19:00 | 小狐狸🦊: 我试试\n",
      "2023-05-18 08:19:00 | 小狐狸🦊: 行\n",
      "2023-05-18 08:18:00 | Andy: （先劝你一起请\n",
      "2023-05-18 08:28:00 | Andy: [图片]\n",
      "2023-05-18 08:19:00 | Andy: [Grimace]\n",
      "2023-05-18 08:19:00 | Andy: 我也这么想的\n",
      "2023-05-18 08:29:00 | 小狐狸🦊: 厉害\n",
      "2023-05-18 08:29:00 | 小狐狸🦊: [图片]\n",
      "2023-05-18 08:41:00 | 小狐狸🦊: 等会来找我\n",
      "2023-05-18 08:41:00 | 小狐狸🦊: 我在510\n",
      "2023-05-18 08:29:00 | Andy: lol\n",
      "2023-05-18 08:45:00 | Andy: 好嘞\n",
      "2023-05-18 12:38:00 | 小狐狸🦊: [图片]\n",
      "2023-05-18 09:21:00 | Andy: 510？\n",
      "2023-05-18 13:05:00 | Andy: [图片]\n",
      "2023-05-18 16:48:00 | 小狐狸🦊: 好 明天见\n",
      "2023-05-18 16:46:00 | Andy: k\n",
      "2023-05-18 16:42:00 | 小狐狸🦊: 算了 还是下次再给你吧 等你不生气的时候TwT\n",
      "2023-05-18 17:06:00 | Andy: 好吧\n",
      "2023-05-18 17:06:00 | Andy: 只是觉得第一个520\n",
      "2023-05-18 17:06:00 | Andy: 前面有点生气\n",
      "2023-05-18 17:06:00 | Andy: 对不起喽\n",
      "2023-05-18 17:09:00 | Andy: 好嘞\n",
      "2023-05-18 17:08:00 | 小狐狸🦊: 没事啦 我只是有点不理解 你也知道我很想和你出去玩 只是还有九天就考试了 考完我们好好去玩okkk?\n",
      "2023-05-18 17:07:00 | Andy: [动画表情]\n",
      "2023-05-18 17:11:00 | Andy: 但廿一宝宝要求可能不一样\n",
      "2023-05-18 17:10:00 | 小狐狸🦊: 你是比较的厉害[CoolGuy]\n",
      "2023-05-18 17:09:00 | Andy: 我没把分看的那么重[Slight]随便考考也有A*\n",
      "2023-05-18 17:11:00 | Andy: nah\n",
      "2023-05-18 17:11:00 | 小狐狸🦊: 我虽然可能maybe考的还行 但是我理解东西所需要的时间比其他人多很多\n",
      "2023-05-18 17:11:00 | Andy: [动画表情]\n",
      "2023-05-18 17:11:00 | Andy: 我要90\n",
      "2023-05-18 17:11:00 | 小狐狸🦊: in order words我笨笨的[CoolGuy]\n",
      "2023-05-18 17:11:00 | Andy: 你要100\n",
      "2023-05-18 17:11:00 | Andy: thats the diff\n",
      "2023-05-18 17:12:00 | 小狐狸🦊: [动画表情]\n",
      "2023-05-18 17:12:00 | Andy: 不打扰喽\n",
      "2023-05-18 17:12:00 | Andy: 你去内卷吧\n",
      "2023-05-18 21:30:00 | Andy: 明天记得多和我拍点照\n",
      "2023-05-18 21:29:00 | Andy: 好嘞！\n",
      "2023-05-18 21:29:00 | 小狐狸🦊: [Kiss][Rose]\n",
      "2023-05-18 21:29:00 | 小狐狸🦊: 明天记得穿polo！！\n",
      "2023-05-18 21:30:00 | Andy: （既然某人过于内卷不愿意出去玩\n",
      "2023-05-18 21:30:00 | 小狐狸🦊: 好！！\n",
      "2023-05-18 21:30:00 | Andy: 那个礼物是啥啊\n",
      "2023-05-18 21:30:00 | Andy: 哦对了\n",
      "2023-05-18 21:31:00 | Andy: but tell me tmr\n",
      "2023-05-18 21:31:00 | Andy: 给点hint\n",
      "2023-05-18 21:30:00 | 小狐狸🦊: 真的想知道吗！！\n",
      "2023-05-18 21:30:00 | Andy: [动画表情]\n",
      "2023-05-18 21:32:00 | Andy: 靠\n",
      "2023-05-18 21:31:00 | 小狐狸🦊: [Drool]\n",
      "2023-05-18 21:31:00 | 小狐狸🦊: or green\n",
      "2023-05-18 21:31:00 | 小狐狸🦊: something not blue\n",
      "2023-05-18 21:32:00 | Andy: 我没听\n",
      "2023-05-18 21:32:00 | Andy: 明天几点到来着\n",
      "2023-05-18 21:32:00 | Andy: miles说\n",
      "2023-05-18 21:33:00 | Andy: o\n",
      "2023-05-18 21:33:00 | 小狐狸🦊: [图片]\n",
      "2023-05-18 21:33:00 | Andy: [动画表情]\n",
      "2023-05-18 21:33:00 | 小狐狸🦊: basketballcourt\n",
      "2023-05-18 21:33:00 | 小狐狸🦊: 7:45吧\n"
     ]
    }
   ],
   "source": [
    "import datetime\n",
    "\n",
    "class Message():\n",
    "    def __init__(self, date, time, author, msg):\n",
    "        self.date = date\n",
    "        self.time = time\n",
    "        self.author = author\n",
    "        self.msg = msg\n",
    "\n",
    "current_date = datetime.date.today()\n",
    "\n",
    "weekdays = {}\n",
    "eng_to_chinese = {\n",
    "    'Monday': '星期一',\n",
    "    'Tuesday': '星期二',\n",
    "    'Wednesday': '星期三',\n",
    "    'Thursday': '星期四',\n",
    "    'Friday': '星期五',\n",
    "    'Saturday': '星期六',\n",
    "    'Sunday': '星期天'\n",
    "}\n",
    "date_format = '%y/%m/%d'\n",
    "\n",
    "for i in range(2,7):\n",
    "    delta = datetime.timedelta(days=i)\n",
    "    date = current_date - delta\n",
    "    weekname = eng_to_chinese[date.strftime('%A')]\n",
    "    weekdays[weekname] = date\n",
    "    \n",
    "# date = weekdays.get(\"weekname\")\n",
    "weekdays[\"今天\"] = current_date\n",
    "weekdays[\"昨天\"] = current_date-datetime.timedelta(days=1)\n",
    "\n",
    "proc_msg = []\n",
    "\n",
    "def printMsg(message):\n",
    "    print(f\"{message.date} {message.time} | {message.author}: {message.msg}\")\n",
    "\n",
    "for msg in all_msg:\n",
    "    time = msg.time\n",
    "    date = weekdays.get(time)\n",
    "    if date is not None:\n",
    "        proc_msg.append(Message(date, \"NA\", msg.author, msg.msg))\n",
    "    elif type(time) == str:\n",
    "        #print(time)\n",
    "        hour, minute = time.split(' ')[0].split(':')\n",
    "        hour = int(hour)\n",
    "        minute = int(minute)\n",
    "        if time.split(' ')[1] == \"pm\" and hour != 12:\n",
    "            hour += 12\n",
    "        #print(hour, minute)\n",
    "        #print(\"---------\")\n",
    "        proc_msg.append(Message(current_date, datetime.time(hour, minute, 0), msg.author, msg.msg))\n",
    "    else:\n",
    "        proc_msg.append(Message(time, \"NA\", msg.author, msg.msg))\n",
    "\n",
    "for msg in proc_msg:\n",
    "    printMsg(msg)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 288,
   "id": "65feb45b",
   "metadata": {},
   "outputs": [
    {
     "ename": "UnicodeEncodeError",
     "evalue": "'gbk' codec can't encode character '\\U0001f98a' in position 17: illegal multibyte sequence",
     "output_type": "error",
     "traceback": [
      "\u001b[1;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[1;31mUnicodeEncodeError\u001b[0m                        Traceback (most recent call last)",
      "Input \u001b[1;32mIn [288]\u001b[0m, in \u001b[0;36m<cell line: 5>\u001b[1;34m()\u001b[0m\n\u001b[0;32m     10\u001b[0m     writer\u001b[38;5;241m.\u001b[39mwriterow(headers)\n\u001b[0;32m     12\u001b[0m     \u001b[38;5;28;01mfor\u001b[39;00m msg \u001b[38;5;129;01min\u001b[39;00m proc_msg:\n\u001b[1;32m---> 13\u001b[0m         \u001b[43mwriter\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mwriterow\u001b[49m\u001b[43m(\u001b[49m\u001b[43m[\u001b[49m\u001b[43mmsg\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mdate\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mmsg\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mtime\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mmsg\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mauthor\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mmsg\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mmsg\u001b[49m\u001b[43m]\u001b[49m\u001b[43m)\u001b[49m\n\u001b[0;32m     15\u001b[0m \u001b[38;5;28mprint\u001b[39m(\u001b[38;5;124mf\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mData exported to \u001b[39m\u001b[38;5;132;01m{\u001b[39;00mfilename\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m successfully.\u001b[39m\u001b[38;5;124m\"\u001b[39m)\n",
      "\u001b[1;31mUnicodeEncodeError\u001b[0m: 'gbk' codec can't encode character '\\U0001f98a' in position 17: illegal multibyte sequence"
     ]
    }
   ],
   "source": [
    "import csv\n",
    "\n",
    "filename = 'output.csv'\n",
    "\n",
    "with open(filename, 'w', encoding='gbk', newline='') as file:\n",
    "    writer = csv.writer(file)\n",
    "\n",
    "    # Write headers\n",
    "    headers = ['Date', 'Time', 'Author', 'Msg']\n",
    "    writer.writerow(headers)\n",
    "    \n",
    "    for msg in proc_msg:\n",
    "        writer.writerow([msg.date, msg.time, msg.author, msg.msg])\n",
    "\n",
    "print(f\"Data exported to {filename} successfully.\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d4ba5cc9",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.9.12"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}