From d61800b7554471cc701bfe85640922b3ecf62fb8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=80=BF=E4=BC=A6=E4=BC=A6?= Date: Wed, 1 Dec 2021 23:44:25 +0800 Subject: [PATCH] close mongo connection --- .vscode/launch.json | 13 +++++++++++++ messages/__pycache__/base.cpython-39.pyc | Bin 1815 -> 1850 bytes .../__pycache__/roomuserseq.cpython-39.pyc | Bin 1105 -> 1183 bytes messages/__pycache__/utils.cpython-39.pyc | Bin 2085 -> 2063 bytes messages/base.py | 2 ++ messages/roomuserseq.py | 2 +- scripts/__pycache__/watcher.cpython-39.pyc | Bin 1509 -> 1493 bytes scripts/__pycache__/webdriver.cpython-39.pyc | Bin 1944 -> 1985 bytes scripts/watcher.py | 1 - scripts/webdriver.py | 3 ++- store/__pycache__/mongo.cpython-39.pyc | Bin 1390 -> 1522 bytes store/mongo.py | 3 +++ 12 files changed, 21 insertions(+), 3 deletions(-) create mode 100644 .vscode/launch.json diff --git a/.vscode/launch.json b/.vscode/launch.json new file mode 100644 index 0000000..76159c2 --- /dev/null +++ b/.vscode/launch.json @@ -0,0 +1,13 @@ +{ + "version": "0.2.0", + "configurations": [ + { + "name": "Python: 当前文件", + "type": "python", + "request": "launch", + "program": "${file}", + "console": "integratedTerminal", + "args": ["https://live.douyin.com/162574996168"] + } + ] +} \ No newline at end of file diff --git a/messages/__pycache__/base.cpython-39.pyc b/messages/__pycache__/base.cpython-39.pyc index 18da0885708b0add7905d32dd7d8bd289c6f2f6c..ac480e6df3dfbf97bf12c36e65deb6f21b674f12 100644 GIT binary patch delta 215 zcmbQvw~LQAk(ZZ?0SJ29mnS~j$h(Kh%z=@Cp@gBCA%#(rA%zKunX{RT#8Mc78744R zNmLyxVQgloWvpRXz_gH|mZ^p*g`t$OC}Z+(CP~JO$)e1$vb$=S7YL_t)i7nTW-=~h zWMn820jd^V$k5DKbY*fsvofQ~a|GB7gaaY0#=-!RW*l%Cwha*X{J zYjRG0aq8p%R!eR>kP8`w7+ILuCQo5CWl~|9e3CVS(Q&d8TQHk8P_9U4ay6R)qt@g# GY}o*=05_fh delta 180 zcmdnRH=U0+k(ZZ?0SG#5mM3oC$h(KhNR5$!p@gBCA%#(rA%zKunX{RT#8Mc78LEzz zFg7#PGS)CGU|Psf%T&XZ!cfXslrs50lO$uxWJ%^&*=;q<3xrd+YM8QEGZ_~$GBT8i z09A`FWN2nAx-fY%vofQ?9RL6T diff --git a/messages/__pycache__/roomuserseq.cpython-39.pyc b/messages/__pycache__/roomuserseq.cpython-39.pyc index b665d8c9263d16f8a3744280d8371e5f810758ff..3eac0d5973272a16ce6af34118bffc3e64a6fa41 100644 GIT binary patch delta 190 zcmcb}F`tt!k(ZZ?0SJU+mM2;;ZR9(}WM%;5mM|<}Oko6)ObZznGNmwyGo&z!Gn6pb zFl4c0u{JXT`7GiLH4F>b7BYx41T$!|PX5KD9a+i`)cbVrlOlqHQ8^mlw{_n-r^`ODM|ygICJuo6LY{!W}sjZ8&G2r a4~XEOe3wyTatw1CqsU}M7V*iqn9BirpFiCI delta 102 zcmbQwd69!Jk(ZZ?0SIF3mM4B<+{kx|Nl6CCEn!%|n8FAonHDlEWJ+NYXQ*LVz`T$_ xoFSM&lX)^9v$m`z%Pp3Y%-qyl9K|I?XX3Am7WsPEG z1j#Yyu;sExv4hzxIUKp1QJjnnsmu$w7BWO}yECM)rm(dzq_8zJMezW6>?s^T9!DyB zGjkO0#tvIXM*hvm7^@i>g(o{Pmof58-pTxhQDyQvmJ}%^pc{%=fCL8<6JrrSkfF(U gi=`wpH+8Za>s%25kPIUbGVw6-F!3;PF!6~20NB_l`Tzg` diff --git a/scripts/__pycache__/webdriver.cpython-39.pyc b/scripts/__pycache__/webdriver.cpython-39.pyc index d5278feebaa7b821750769dd0030f32394f24399..eb223492e507e3542e7215ad4be8c44bdf63e2e1 100644 GIT binary patch delta 438 zcmY+A%}T>S5XWbdY}&-8>35s@eUMr(phpoe;>C-X9>kl)twR3ZliN4q_Gt!FJoMiPQEH|m-HU6i~sF&LM0iP3K`Tzg` delta 357 zcmX@eKZBn)k(ZZ?0SLlWmM2DS!Yq3nWrF zCmS*e*K+|);|5|LNrqa^66t1!TCN(d1u`kTU=z3&$kuSC@JTWNNq#WN3sfrrX9>bt zLO^k0AQpj(h{9Q7KymRo5ZA%nB$pxqa+4Owns9~`1{NS}W@co_6Eb9&z*wAQz`zKG zkw7wqA(%l^a`IOuh0Ti0oQ#a}o3&Y(87G^vmNKeMUd`&n0ELUxPzm+PQ`vO@IE_f$ diff --git a/scripts/watcher.py b/scripts/watcher.py index 5623741..5c4beb6 100644 --- a/scripts/watcher.py +++ b/scripts/watcher.py @@ -1,4 +1,3 @@ -import time import concurrent.futures import queue diff --git a/scripts/webdriver.py b/scripts/webdriver.py index 9a49a00..6aeb69a 100644 --- a/scripts/webdriver.py +++ b/scripts/webdriver.py @@ -15,7 +15,7 @@ from store.mongo import MongoStore def go(url): chrome_options = Options() chrome_options.add_argument('--proxy-server=%s' % config()['webdriver']['proxy']) - # chrome_options.add_argument('--headless') + chrome_options.add_argument('--headless') proxy = Proxy() proxy.proxy_type = ProxyType.MANUAL @@ -52,6 +52,7 @@ def go(url): 'admin_user_ids': roomInfo['room']['admin_user_ids'], 'owner': roomInfo['room']['owner'] }) + store.close() wait.until(presence_of_element_located((By.CLASS_NAME, "oSu9Aw19"))) \ No newline at end of file diff --git a/store/__pycache__/mongo.cpython-39.pyc b/store/__pycache__/mongo.cpython-39.pyc index 84ea61386818d04382e4c00d5e8259db5945e584..7e2817fd3b04a73c4110ed5fff3667fb918847ad 100644 GIT binary patch delta 678 zcmZuvJ8K(35Z=9qbh^`>Bs*3Tk|u!^O5uR32!b1j6vlF+PK|Qhu7t$tHte1t2?k?O z5lAJrOBGkSyq3Hlc35vYm)&d8Mfq?d#eM*5_nryWaHA#U6x6^hFm_%iqlb@UnlxU51ZLccb9 zA{&c-E+pSCyOJ`2Nf3QQU%Zz%3#dIX#!p0=@+6;#3_{2xn+X4pF_Gf=a9tYet~ENs zLK8^-DUB086dCV0C4;FYB4ZQeG!BdNSnh06%3X*x^{v%=yV|Fu^CuacG)J~#wa?y` z<~GE3_1Hc+__sM&HP7Pgi+sGL?Fq!%`nNsn;;=y@5%Vk`GLg(@Jj>O?4LsUxH5LAO zqH^_Ae|KI|U;lDm9fw$k4l^vWl*A>o3SZ|+gg@oe38_g`CRWF`>U?8@v4rWEo6%XIkx z$q)Dy7B)6pTU+=A-W!7v%)p#GoH^&7JM-myGC%Y@roqY&pTl=O_a*7%Pn)|HUgG5m z=~|p}d!ltqoN)(gnY-MBO1Xcd1=aj@>IS4RQxOejnq*3nJ9VM&pUud1pi2sk3)WAf zNc6K|5(8xeC4B^p{Q(DMek!uwQp(heKB%LxaHP18!oKJwu?S3Qt0SY{s{Xp?eXxaG z1MpPOxH|FR1YvVK8^)=S8MnH9HIQD2&IlzQWc3t_GP+H8yXF zuinh&*8Gj@|HVZ~JPy+=Z76DO6I<~9lr4niA1z;Lz=Z`nMJpxL ZR8#Aqy+T>g(^0}7BGDlzH&L#;`ZreSVg&#I diff --git a/store/mongo.py b/store/mongo.py index df85009..52116ed 100644 --- a/store/mongo.py +++ b/store/mongo.py @@ -7,6 +7,9 @@ class MongoStore: self.client = pymongo.MongoClient(config()['mongo']['uri']) self.db = self.client[config()['mongo']['dbname']] + def close(self): + self.client.close() + def set_collection(self, collection): self.collection = self.db[collection]