from multiprocessing import Process

from sqlalchemy import create_engine, pool

from tempfile import mkdtemp

import shutil

from iro.model.utils import WithSession, POOL_SIZE as DB_POOL_SIZE

import iro.model.user as imuser

from iro.model.schema import User, Base

from ngdatabase.mysql import Server, createConfig, Database

class SampleDatabase(Database):

    def createPassword(self):

        self.password="test"

        return self.password

from twisted.python import log

import logging

logging.basicConfig(level=logging.DEBUG, format='%(asctime)s %(name)s(%(processName)s)-%(levelname)s: %(message)s')

#observer = log.PythonLoggingObserver()

#observer.start()

def xxxxx(userid):

    import xmlrpclib

    return xmlrpclib.ServerProxy('http://localhost:7080/RPC2').listMethods()

def main():

    from twisted.internet import reactor

    from twisted.web import xmlrpc, server

    from iro.view.xmlrpc import getResource

    class XMLRPCDebug(xmlrpc.XMLRPC): 

        def xmlrpc_stop(self):

            reactor.callLater(0.5,reactor.stop)

            return ""

    root=getResource()

    root.putChild('debug', XMLRPCDebug())

    reactor.listenTCP(7080, server.Site(root))

    logging.info("Server is running now...")

    reactor.run()

if __name__ == '__main__':

    tdir = mkdtemp(prefix='iro-mysql-')

    try:

        with open('%s/my.cnf'%tdir,'w') as cnf:

            cnf.write(createConfig(tdir))

        s = Server('%s/my.cnf'%tdir)

        s.create()

        s.start()

        d=SampleDatabase("test","test",'%s/my.cnf'%tdir)

        d.create()

        engine = create_engine('mysql://test:test@localhost/test?unix_socket=%s/socket'%tdir,

               poolclass = pool.SingletonThreadPool,  pool_size=DB_POOL_SIZE, )

        imuser.setEngine(engine)

        try:

            Base.metadata.create_all(engine)

            with WithSession(engine, autocommit=True) as session:

                session.add(User(name='test',apikey='abcdef123456789'))

            p = Process(target=main)

            p.start()

            p.join()

        finally:

            s.stop()

    finally:

        shutil.rmtree(tdir)