Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Sign in / Register
Toggle navigation
S
shadowban-eu-backend
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Locked Files
Issues
0
Issues
0
List
Boards
Labels
Service Desk
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Security & Compliance
Security & Compliance
Dependency List
License Compliance
Packages
Packages
List
Container Registry
Analytics
Analytics
CI / CD
Code Review
Insights
Issues
Repository
Value Stream
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
nanahira
shadowban-eu-backend
Commits
e096cece
Commit
e096cece
authored
Dec 15, 2020
by
Raphael Beer
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Fix: open sessions/connections on exit
parent
4d04569f
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
15 additions
and
0 deletions
+15
-0
backend.py
backend.py
+12
-0
db.py
db.py
+3
-0
No files found.
backend.py
View file @
e096cece
...
@@ -154,6 +154,16 @@ if args.debug is not None:
...
@@ -154,6 +154,16 @@ if args.debug is not None:
ensure_dir
(
debug_dir
)
ensure_dir
(
debug_dir
)
debug_file
=
open
(
args
.
debug
,
"a"
)
debug_file
=
open
(
args
.
debug
,
"a"
)
async
def
close_sessions
(
app
):
print
(
"
\n
Closing
%
s guest sessions"
%
len
(
TwitterSession
.
guest_sessions
))
for
session
in
TwitterSession
.
guest_sessions
:
await
session
.
close
()
async
def
close_database
(
app
):
global
db
print
(
"Closing database connection"
)
db
.
close
()
def
run
():
def
run
():
global
db
global
db
db
=
connect
(
db
=
connect
(
...
@@ -167,6 +177,8 @@ def run():
...
@@ -167,6 +177,8 @@ def run():
loop
.
run_until_complete
(
login_guests
())
loop
.
run_until_complete
(
login_guests
())
app
=
web
.
Application
()
app
=
web
.
Application
()
app
.
add_routes
(
routes
)
app
.
add_routes
(
routes
)
app
.
on_shutdown
.
append
(
close_sessions
)
app
.
on_cleanup
.
append
(
close_database
)
web
.
run_app
(
app
,
host
=
args
.
host
,
port
=
args
.
port
)
web
.
run_app
(
app
,
host
=
args
.
host
,
port
=
args
.
port
)
if
args
.
daemon
:
if
args
.
daemon
:
...
...
db.py
View file @
e096cece
...
@@ -35,6 +35,9 @@ class Database:
...
@@ -35,6 +35,9 @@ class Database:
def
get_result_by_screen_name
(
self
,
screen_name
):
def
get_result_by_screen_name
(
self
,
screen_name
):
return
self
.
results
.
find_one
({
"profile.screen_name"
:
screen_name
},
sort
=
[(
"_id"
,
DESCENDING
)],
projection
=
{
"_id"
:
False
})
return
self
.
results
.
find_one
({
"profile.screen_name"
:
screen_name
},
sort
=
[(
"_id"
,
DESCENDING
)],
projection
=
{
"_id"
:
False
})
def
close
(
self
):
self
.
client
.
close
()
def
connect
(
host
=
None
,
port
=
27017
,
db
=
'tester'
,
username
=
None
,
password
=
None
):
def
connect
(
host
=
None
,
port
=
27017
,
db
=
'tester'
,
username
=
None
,
password
=
None
):
if
host
is
None
:
if
host
is
None
:
raise
ValueError
(
'[mongoDB] Database constructor needs a `host`name or ip!'
)
raise
ValueError
(
'[mongoDB] Database constructor needs a `host`name or ip!'
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment