gsd
/
LinuxGSM
mirror of https://github.com/GameServerManagers/LinuxGSM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
Browse Source

fix: replace deprecated optparse with argparse (#3913)

pull/3963/head
eggplants 3 years ago
committed by GitHub
parent
dc420668a2
commit
81998f73c7
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 72 additions and 67 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 139
      lgsm/functions/query_gsquery.py

139
lgsm/functions/query_gsquery.py
View File

@ -6,66 +6,66 @@
# Website: https://linuxgsm.com
# Description: Allows querying of various game servers.
import optparse
import argparse
import socket
import sys
engine_types=('protocol-valve','protocol-quake3', 'protocol-quake3','protocol-gamespy1','protocol-unreal2','ut3 minecraft','minecraftbe','jc2mp','mumbleping','soldat','teeworlds')
class gsquery:
def __init__(self, options, arguments):
self.option = options
server_response_timeout = 5
default_buffer_length = 1024
sourcequery=('protocol-valve','avalanche3.0','barotrauma','madness','quakelive','realvirtuality','refractor','source','goldsrc','spark','starbound','unity3d','unreal4','wurm')
idtech2query=('protocol-quake3','idtech2','quake','iw2.0')
idtech3query=('protocol-quake3','iw3.0','ioquake3','qfusion')
minecraftquery=('minecraft','lwjgl2')
minecraftbequery=('minecraftbe',)
jc2mpquery=('jc2mp',)
mumblequery=('mumbleping',)
soldatquery=('soldat',)
twquery=('teeworlds',)
unrealquery=('protocol-gamespy1','unreal')
unreal2query=('protocol-unreal2','unreal2')
unreal3query=('ut3','unreal3')
def __init__(self, arguments):
self.argument = arguments
#
self.server_response_timeout = 5
self.default_buffer_length = 1024
#
sourcequery=['protocol-valve','avalanche3.0','barotrauma','madness','quakelive','realvirtuality','refractor','source','goldsrc','spark','starbound','unity3d','unreal4','wurm']
idtech2query=['protocol-quake3','idtech2','quake','iw2.0']
idtech3query=['protocol-quake3','iw3.0','ioquake3','qfusion']
minecraftquery=['minecraft','lwjgl2']
minecraftbequery=['minecraftbe']
jc2mpquery=['jc2mp']
mumblequery=['mumbleping']
soldatquery=['soldat']
twquery=['teeworlds']
unrealquery=['protocol-gamespy1','unreal']
unreal2query=['protocol-unreal2','unreal2']
unreal3query=['ut3','unreal3']
if self.option.engine in sourcequery:
if self.argument.engine in self.sourcequery:
self.query_prompt_string = b'\xFF\xFF\xFF\xFFTSource Engine Query\0'
elif self.option.engine in idtech2query:
elif self.argument.engine in self.idtech2query:
self.query_prompt_string = b'\xff\xff\xff\xffstatus\x00'
elif self.option.engine in idtech3query:
elif self.argument.engine in self.idtech3query:
self.query_prompt_string = b'\xff\xff\xff\xffgetstatus'
elif self.option.engine in jc2mpquery:
elif self.argument.engine in self.jc2mpquery:
self.query_prompt_string = b'\xFE\xFD\x09\x10\x20\x30\x40'
elif self.option.engine in minecraftquery:
elif self.argument.engine in self.minecraftquery:
self.query_prompt_string = b'\xFE\xFD\x09\x3d\x54\x1f\x93'
elif self.option.engine in minecraftbequery:
elif self.argument.engine in self.minecraftbequery:
self.query_prompt_string = b'\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\xff\xff\x00\xfe\xfe\xfe\xfe\xfd\xfd\xfd\xfd\x12\x34\x56\x78\x00\x00\x00\x00\x00\x00\x00\x00'
elif self.option.engine in mumblequery:
elif self.argument.engine in self.mumblequery:
self.query_prompt_string = b'\x00\x00\x00\x00\x01\x02\x03\x04\x05\x06\x07\x08'
elif self.option.engine in soldatquery:
elif self.argument.engine in self.soldatquery:
self.query_prompt_string = b'\x69\x00'
elif self.option.engine in twquery:
self.query_prompt_string = b"\x04\x00\x00\xff\xff\xff\xff\x05" + bytearray(511)
elif self.option.engine in unrealquery:
elif self.argument.engine in self.twquery:
self.query_prompt_string = b'\x04\x00\x00\xff\xff\xff\xff\x05' + bytearray(511)
elif self.argument.engine in self.unrealquery:
self.query_prompt_string = b'\x5C\x69\x6E\x66\x6F\x5C'
elif self.option.engine in unreal2query:
elif self.argument.engine in self.unreal2query:
self.query_prompt_string = b'\x79\x00\x00\x00\x00'
elif self.option.engine in unreal3query:
elif self.argument.engine in self.unreal3query:
self.query_prompt_string = b'\xFE\xFD\x09\x00\x00\x00\x00'
self.connected = False
self.response = None
self.sanity_checks()
@staticmethod
def fatal_error(self, error_message, error_code=1):
def fatal_error(error_message, error_code=1):
sys.stderr.write('ERROR: ' + str(error_message) + '\n')
sys.exit(error_code)
@staticmethod
def exit_success(self, success_message=''):
def exit_success(success_message=''):
sys.stdout.write('OK: ' + str(success_message) + '\n')
sys.exit(0)
@ -74,10 +74,10 @@ class gsquery:
connection = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
connection.settimeout(self.server_response_timeout)
try:
self.connected = connection.connect((self.option.address, int(self.option.port)))
self.connected = connection.connect((self.argument.address, int(self.argument.port)))
except socket.timeout:
self.fatal_error('Request timed out', 1)
except:
except Exception:
self.fatal_error('Unable to connect', 1)
# Send.
connection.send(self.query_prompt_string)
@ -95,52 +95,57 @@ class gsquery:
else:
self.exit_success(str(self.response))
def sanity_checks(self):
if not self.option.address:
self.fatal_error('No IPv4 address supplied.', 4)
if not self.option.port:
self.fatal_error('No port supplied.', 4)
if __name__ == '__main__':
parser = optparse.OptionParser(
usage='usage: python3 %prog [options]',
version='%prog 0.0.1'
def parse_args():
parser = argparse.ArgumentParser(
description='Allows querying of various game servers.',
usage='usage: python3 %(prog)s [options]',
add_help=False
)
parser.add_option(
parser.add_argument(
'-a', '--address',
action='store',
dest='address',
default=False,
type=str,
required=True,
help='The IPv4 address of the server.'
)
parser.add_option(
parser.add_argument(
'-p', '--port',
action='store',
dest='port',
default=False,
type=int,
required=True,
help='The IPv4 port of the server.'
)
parser.add_option(
parser.add_argument(
'-e', '--engine',
action='store',
dest='engine',
default=False,
help='Engine type: protocol-valve protocol-quake3 protocol-quake3 protocol-gamespy1 protocol-unreal2 ut3 minecraft minecraftbe jc2mp mumbleping soldat teeworlds'
metavar='ENGINE',
choices=engine_types,
help='Engine type: ' + ' '.join(engine_types)
)
parser.add_option(
parser.add_argument(
'-v', '--verbose',
action='store_true',
dest='verbose',
default=False,
help='Display verbose output.'
)
parser.add_option(
parser.add_argument(
'-d', '--debug',
action='store_true',
dest='debug',
default=False,
help='Display debugging output.'
)
options, arguments = parser.parse_args()
server = gsquery(options, arguments)
parser.add_argument(
'-V', '--version',
action='version',
version='%(prog)s 0.0.1',
help='Display version and exit.'
)
parser.add_argument(
'-h', '--help',
action='help',
help='Display help and exit.'
)
return parser.parse_args()
def main():
arguments = parse_args()
server = gsquery(arguments)
server.responding()
if __name__ == '__main__':
main()

Write Preview
Loading…
Cancel
Save