Show More
@@ -39,6 +39,7 b' import socket' | |||
|
39 | 39 | import errno |
|
40 | 40 | import random |
|
41 | 41 | from functools import update_wrapper, partial, wraps |
|
42 | from contextlib import closing | |
|
42 | 43 | |
|
43 | 44 | import pygments.lexers |
|
44 | 45 | import sqlalchemy |
@@ -1190,23 +1191,21 b' def user_agent_normalizer(user_agent_raw' | |||
|
1190 | 1191 | return ua |
|
1191 | 1192 | |
|
1192 | 1193 | |
|
1193 | def get_available_port(min_port=40000, max_port=55555): | |
|
1194 | sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) | |
|
1195 | hostname = '127.0.0.1' | |
|
1196 | pick_port = min_port | |
|
1197 | ||
|
1194 | def get_available_port(min_port=40000, max_port=55555, use_range=False): | |
|
1195 | hostname = '' | |
|
1198 | 1196 | for _ in range(min_port, max_port): |
|
1197 | pick_port = 0 | |
|
1198 | if use_range: | |
|
1199 | 1199 | pick_port = random.randint(min_port, max_port) |
|
1200 | ||
|
1201 | with closing(socket.socket(socket.AF_INET, socket.SOCK_STREAM)) as s: | |
|
1200 | 1202 | try: |
|
1201 |
s |
|
|
1202 | sock.close() | |
|
1203 | break | |
|
1203 | s.bind((hostname, pick_port)) | |
|
1204 | s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) | |
|
1205 | return s.getsockname()[1] | |
|
1204 | 1206 | except OSError: |
|
1205 | 1207 | continue |
|
1206 | 1208 | except socket.error as e: |
|
1207 | 1209 | if e.args[0] in [errno.EADDRINUSE, errno.ECONNREFUSED]: |
|
1208 | 1210 | continue |
|
1209 | 1211 | raise |
|
1210 | ||
|
1211 | del sock | |
|
1212 | return pick_port |
General Comments 0
You need to be logged in to leave comments.
Login now