From cf904073c46a6253d79b1bd4484ccaf2f8a4e57f Mon Sep 17 00:00:00 2001
From: ttdevs <ttdevs@gmail.com>
Date: Fri, 4 Jan 2019 11:00:10 +0800
Subject: [PATCH] change config file(cdn_list, station_name.txt) load manage

---
 agency/cdn_utils.py        | 4 +++-
 init/select_ticket_info.py | 4 +++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/agency/cdn_utils.py b/agency/cdn_utils.py
index 7626f6f..f081a6e 100755
--- a/agency/cdn_utils.py
+++ b/agency/cdn_utils.py
@@ -1,6 +1,7 @@
 # encoding=utf8
 import collections
 import json
+import os
 import re
 import sys
 import csv
@@ -53,7 +54,8 @@ class CDNProxy:
     def open_cdn_file(self):
         cdn = []
         cdn_re = re.compile("CONNECT (\S+) HTTP/1.1")
-        with open("./cdn_list", "r") as f:
+        path = os.path.join(os.path.dirname(__file__), '../cdn_list')
+        with open(path, "r") as f:
             for i in f.readlines():
                 # print(i.replace("\n", ""))
                 cdn_list = re.findall(cdn_re, i)
diff --git a/init/select_ticket_info.py b/init/select_ticket_info.py
index 054cc87..abe14c5 100755
--- a/init/select_ticket_info.py
+++ b/init/select_ticket_info.py
@@ -1,6 +1,7 @@
 # -*- coding=utf-8 -*-
 import datetime
 import random
+import os
 import socket
 import sys
 import threading
@@ -118,7 +119,8 @@ class select:
         :param station:
         :return:
         """
-        result = open('station_name.txt')
+        path = os.path.join(os.path.dirname(__file__), '../station_name.txt')
+        result = open(path)
         info = result.read().split('=')[1].strip("'").split('@')
         del info[0]
         station_name = {}