server.py

from __future__ import print_function
from __future__ import absolute_import
from builtins import str
from SimpleWebSocketServer import SimpleWebSocketServer, WebSocket
import json
from .SPaircraft import Mission
from .saveSol import gendes, gencsm
from shutil import copyfile

from .subs.optimalD8 import get_optimalD8_subs
from .aircraft import Mission
from .SPaircraft import optimize_aircraft

EXIT = [False]
ID = 0
LASTSOL = [None]


def genfiles(m, sol):
    global ID
    gensoltxt(m, sol, ID)
    gencsm(m, sol, 'optimalD8', ID)
    copyfile("d82.csm", "d82_%03i.csm" % ID)
    ID += 1


def gensoltxt(m, sol, ID):
    with open("sol_%03i.txt" % ID, "w") as f:
        for var in sorted(m.varkeys, key=str):
            f.write("%s [%s]\t\t%f\n" % (var, var.unitstr(options=":~",
                                                          dimless="-"),
                                         sol["variables"][var]))


class SPaircraftServer(WebSocket):

    def handleMessage(self):
        print("< received", repr(self.data))
        try:
            self.data = json.loads(self.data)
            print(self.data)

            config = 'optimalD8'
            substitutions = get_optimalD8_subs()
            fixedBPR = False
            pRatOpt = True
            mutategparg = True
            m = Mission(3, 2, config, 1)
            m.cost = m['W_{f_{total}}']

            for name, value in list(self.data.items()):
                try:
                    key = m.design_parameters[name]
                    m.substitutions[key] = value
                except KeyError as e:
                    print(repr(e))

            sol = optimize_aircraft(m, substitutions, fixedBPR, pRatOpt, mutategparg)
            LASTSOL[0] = sol
            genfiles(m, sol)

            self.send({"status": "optimal",
                       "msg": ("Successfully optimized."
                               " Optimal heat transfer: %.1f watts "
                               % sol["variables"][m.Q])})
        except Exception as e:
            self.send({"status": "unknown", "msg": "The last solution"
                      " raised an exception; tweak it and send again."})
            print(type(e), e)

    def send(self, msg):
        print("> sent", repr(msg))
        self.sendMessage(str(json.dumps(msg)))

    def handleConnected(self):
        print(self.address, "connected")

    def handleClose(self):
        print(self.address, "closed")
        EXIT[0] = True


if __name__ == "__main__":
    objective = 'W_{f_{total}}'
    aircraft = 'optimald8'
    substitutions = get_optimalD8_subs()
    fixedBPR = False
    pRatOpt = True
    mutategparg = True
    LASTSOL[0] = None
    sol = optimize_aircraft(objective, aircraft, substitutions, fixedBPR, pRatOpt, mutategparg, x0 = LASTSOL[0])
    LASTSOL[0] = sol
    genfiles(m, sol)
    server = SimpleWebSocketServer('', 8000, SPaircraftServer)
    while not EXIT[0]:
        server.serveonce()
    print("Python server has exited.")