diff --git a/5_RemoteDesktop/server.py b/5_RemoteDesktop/server.py index 1c58ee5..f8faee0 100755 --- a/5_RemoteDesktop/server.py +++ b/5_RemoteDesktop/server.py @@ -1,8 +1,15 @@ #!/usr/bin/env python +import sys +PY3_OR_LATER = sys.version_info[0] >= 3 + import ctypes from flask import Flask, request, send_file from PIL import ImageGrab -from StringIO import StringIO + +if PY3_OR_LATER: + from io import BytesIO as StreamIO +else: + from io import StringIO as StreamIO # http://msdn.microsoft.com/en-us/library/windows/desktop/ms646260%28v=vs.85%29.aspx MOUSEEVENTF_LEFTDOWN = 2