""
文件讀取。YamlReader讀取yaml文件
"""
import yaml
import os
from xlrd import open_workbook
class YamlReader:
 def __init__(self, yamlf):
  if os.path.exists(yamlf):
   self.yamlf = yamlf
  else:
   raise FileNotFoundError('文件不存在！')
  self._data = None
 @property
 def data(self):
  # 如果是第一次調(diào)用data，讀取yaml文檔，否則直接返回之前保存的數(shù)據(jù)
  if not self._data:
   with open(self.yamlf, 'rb') as f:
    self._data = list(yaml.safe_load_all(f)) # load后是個(gè)generator，用list組織成列表
  return self._data

"""
讀取配置。這里配置文件用的yaml，也可用其他如XML,INI等，需在file_reader中添加相應(yīng)的Reader進(jìn)行處理。
"""
import os
from util_s.file_reader import YamlReader
 
# 通過(guò)當(dāng)前文件的絕對(duì)路徑，其父級(jí)目錄一定是框架的base目錄，然后確定各層的絕對(duì)路徑。如果你的結(jié)構(gòu)不同，可自行修改。
# 之前直接拼接的路徑，修改了一下，用現(xiàn)在下面這種方法，可以支持linux和windows等不同的平臺(tái)，也建議大家多用os.path.split()和os.path.join()，不要直接+'\\xxx\\ss'這樣
BASE_PATH = os.path.split(os.path.dirname(os.path.abspath(__file__)))[0]
CONFIG_FILE = os.path.join(BASE_PATH, 'Config', 'config.yml')
GF_CONFIG_FILE = os.path.join(BASE_PATH, 'Config', 'gf_config.yml')
CM_CONFIG_FILE = os.path.join(BASE_PATH, 'Config', 'cm_config.yml')
APK_CONFIG_FILE = os.path.join(BASE_PATH, 'Config', 'apk_config.yml')
 
class Config:
 def __init__(self, config=""):
  if config == "gf":
   self.config = YamlReader(GF_CONFIG_FILE).data
  if config == "cm":
   self.config = YamlReader(CM_CONFIG_FILE).data
  if config == "apk":
   self.config = YamlReader(APK_CONFIG_FILE).data
  else:
   self.config = YamlReader(CONFIG_FILE).data
 
 def get(self, element, index=0):
  """
  yaml是可以通過(guò)'---'分節(jié)的。用YamlReader讀取返回的是一個(gè)list，第一項(xiàng)是默認(rèn)的節(jié)，如果有多個(gè)節(jié)，可以傳入index來(lái)獲取。
  這樣我們其實(shí)可以把框架相關(guān)的配置放在默認(rèn)節(jié)，其他的關(guān)于項(xiàng)目的配置放在其他節(jié)中?？梢栽诳蚣苤袑?shí)現(xiàn)多個(gè)項(xiàng)目的測(cè)試。
  """
  return self.config[index].get(element)

from util_s.config import Config
 
config = Config("gf")
mail_host = config.get("mail_host")
my_sender = config.get("my_sender") # 發(fā)件人郵箱賬號(hào)
my_pass = config.get("my_pass") # 發(fā)件人郵箱密碼
my_user = config.get("my_user") # 收件人郵箱賬號(hào)，我這邊發(fā)送給自己