60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
|
import base64
|
||
|
|
import qrcode
|
||
|
|
import argparse
|
||
|
|
import flask; from flask import request
|
||
|
|
import socket
|
||
|
|
import sys
|
||
|
|
import io
|
||
|
|
|
||
|
|
def localwhoami(port):
|
||
|
|
site = flask.Flask(__name__)
|
||
|
|
my_ip = socket.gethostbyname(socket.gethostname())
|
||
|
|
print(my_ip)
|
||
|
|
my_url = f'http://{my_ip}:{port}'
|
||
|
|
url_qr = qrcode.make(my_url, border=0)
|
||
|
|
png_bytes = io.BytesIO()
|
||
|
|
url_qr.save(png_bytes, format='png')
|
||
|
|
png_bytes.seek(0)
|
||
|
|
png_bytes = png_bytes.read()
|
||
|
|
png_base64 = base64.b64encode(png_bytes).decode('ascii')
|
||
|
|
|
||
|
|
@site.route('/')
|
||
|
|
def root():
|
||
|
|
ip = request.remote_addr
|
||
|
|
if ip == '127.0.0.1':
|
||
|
|
ip = my_ip
|
||
|
|
return f'''
|
||
|
|
<meta charset="UTF-8">
|
||
|
|
<meta name="viewport" content="width=device-width, initial-scale=1.0"/>
|
||
|
|
<style>
|
||
|
|
html{{height: 100vh;box-sizing: border-box;}}
|
||
|
|
*, *:before, *:after{{box-sizing: inherit;}}
|
||
|
|
body{{display:grid;grid-auto-flow:row;grid-gap:16px;justify-content:center;background-color:grey}}
|
||
|
|
div{{background-color: white; padding: 16px; border-radius: 16px;}}
|
||
|
|
</style>
|
||
|
|
<div>
|
||
|
|
<center>You are</center>
|
||
|
|
<center style="font-size:32pt; font-weight:bold">{ip}</center>
|
||
|
|
</div>
|
||
|
|
<div>
|
||
|
|
<div><img src="data:image/png;base64,{png_base64}"/></div>
|
||
|
|
<center>{my_url}</center>
|
||
|
|
</div>
|
||
|
|
'''
|
||
|
|
site.run(host='0.0.0.0', port=port)
|
||
|
|
|
||
|
|
def localwhoami_argparse(args):
|
||
|
|
return localwhoami(port=args.port)
|
||
|
|
|
||
|
|
def main(argv):
|
||
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
||
|
|
|
||
|
|
parser.add_argument('port')
|
||
|
|
parser.set_defaults(func=localwhoami_argparse)
|
||
|
|
|
||
|
|
args = parser.parse_args(argv)
|
||
|
|
return args.func(args)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
raise SystemExit(main(sys.argv[1:]))
|