auto get realpath

server.py

@@ -1,10 +1,28 @@
 #!/usr/bin/python
+# -*- coding: utf-8 -*-
+#
+# Copyright 2015 breakwall
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License. You may obtain
+# a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+
 import time
 import sys
 import threading
 import os
 
-os.chdir(os.path.dirname(os.path.realpath(__file__)))
+if __name__ == '__main__':
+	import inspect
+	os.chdir(os.path.dirname(os.path.realpath(inspect.getfile(inspect.currentframe()))))
 
 import server_pool
 import db_transfer

shadowsocks/asyncdns.py

@@ -26,10 +26,10 @@ import logging
 
 if __name__ == '__main__':
     import sys
-    try:
-        sys.path.insert(0, os.path.join(os.path.dirname(__file__), '../'))
-    except:
-        sys.path.insert(0, os.path.join(sys.path[0], '../'))
+    import inspect
+    file_path = os.path.dirname(os.path.realpath(inspect.getfile(inspect.currentframe())))
+    os.chdir(file_path)
+    sys.path.insert(0, os.path.join(file_path, '../'))
 
 from shadowsocks import common, lru_cache, eventloop, shell
 

shadowsocks/local.py

@@ -24,12 +24,10 @@ import logging
 import signal
 
 if __name__ == '__main__':
-    try:
-        os.chdir(os.path.dirname(os.path.realpath(__file__)))
-        sys.path.insert(0, os.path.join(os.path.dirname(__file__), '../'))
-    except:
-        os.chdir(os.path.dirname(sys.path[0]))
-        sys.path.insert(0, os.path.join(sys.path[0], '../'))
+    import inspect
+    file_path = os.path.dirname(os.path.realpath(inspect.getfile(inspect.currentframe())))
+    os.chdir(file_path)
+    sys.path.insert(0, os.path.join(file_path, '../'))
 
 from shadowsocks import shell, daemon, eventloop, tcprelay, udprelay, asyncdns
 

shadowsocks/server.py

@@ -24,12 +24,10 @@ import logging
 import signal
 
 if __name__ == '__main__':
-    try:
-        os.chdir(os.path.dirname(os.path.realpath(__file__)))
-        sys.path.insert(0, os.path.join(os.path.dirname(__file__), '../'))
-    except:
-        os.chdir(os.path.dirname(sys.path[0]))
-        sys.path.insert(0, os.path.join(sys.path[0], '../'))
+    import inspect
+    file_path = os.path.dirname(os.path.realpath(inspect.getfile(inspect.currentframe())))
+    os.chdir(file_path)
+    sys.path.insert(0, os.path.join(file_path, '../'))
 
 from shadowsocks import shell, daemon, eventloop, tcprelay, udprelay, \
     asyncdns, manager