开机自动获取spark用户名和服务器

Posted 2020-10-15 hjfjesse

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用户名和服务器