mirrors/py-kms
1
0
Fork 0
mirror of https://github.com/Py-KMS-Organization/py-kms.git synced 2024-09-19 07:18:51 -04:00
Code Issues Projects Releases Packages Wiki Activity

Block SQLITE in case sqlite-web files are missing

Browse source
This commit is contained in:
Simon Beginn 2021-10-21 13:22:03 +02:00
parent 4ee742d671
commit eb42224bf1
No known key found for this signature in database
GPG key ID: 033A4D4CE4E063D6
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
docker/entrypoint.py
View file

@ -17,6 +17,7 @@ argumentVariableMapping = {
'-S': 'LOGSIZE',
'-e': 'EPID'
}
sqliteWebPath = '/home/sqlite_web/sqlite_web.py'
# Build the command to execute
listenIP = os.environ.get('IP', '0.0.0.0')
@ -27,7 +28,7 @@ for (arg, env) in argumentVariableMapping.items():
command.append(arg)
command.append(os.environ.get(env))
enableSQLITE = os.environ.get('SQLITE', 'false').lower() == 'true'
enableSQLITE = os.path.isfile(sqliteWebPath) and os.environ.get('SQLITE', 'false').lower() == 'true'
if enableSQLITE:
print('Storing database file to ' + dbPath)
dbPath = os.path.join('db', 'pykms_database.db')
@ -43,7 +44,7 @@ if enableSQLITE:
if not os.path.isfile(dbPath):
# Start a dummy activation to ensure the database file is created
subprocess.run(['/usr/bin/python3', 'pykms_Client.py', listenIP, listenPort, '-m', 'Windows10', '-n', 'DummyClient', '-c', 'ae3a27d1-b73a-4734-9878-70c949815218'])
sqliteProcess = subprocess.Popen(['/usr/bin/python3', '/home/sqlite_web/sqlite_web.py', '-H', listenIP, '--read-only', '-x', dbPath, '-p', os.environ.get('SQLITE_PORT', 8080)])
sqliteProcess = subprocess.Popen(['/usr/bin/python3', sqliteWebPath, '-H', listenIP, '--read-only', '-x', dbPath, '-p', os.environ.get('SQLITE_PORT', 8080)])
try:
pykmsProcess.wait()