1. 程式人生 > >開機自動獲取spark用戶名和服務器

開機自動獲取spark用戶名和服務器

pda 添加 img system one password class color b-

技術分享圖片
import os.path
import getpass
import platform
import time

username = getpass.getuser()   #獲取當前用戶名
home = os.path.expanduser(~)  #獲取用戶目錄
OS = platform.system()      #獲取系統類型

if OS == "Windows":
    spark_home = %s/AppData/Roaming/Spark % home
else:
    spark_home = %s/.Spark % home
spark_config 
= %s/spark.properties % spark_home def add_account(): account_info = """ username = %s server=chat.base-fx.com passwordSaved=true """ % username f = open(spark_config, w) f.write(account_info) if os.path.exists(spark_config):  #如果配置目錄存在 f = open(spark_config, r) content = f.read()
if username not in content: add_account() else: os.makedirs(spark_home)   #不存在創建目錄並添加內容 add_account()
View Code

開機自動獲取spark用戶名和服務器

開機自動獲取spark用戶名和服務器