'util/init.d/weewx.debian',

'util/init.d/weewx.lsb',

'util/init.d/weewx.redhat',

"""Specialized version of install, which adds a --no-prompt option to

the 'install' command."""

# Add an option for --no-prompt:

user_options = install.user_options + [('no-prompt', None, 'Do not prompt for station info')]

def initialize_options(self, *args, **kwargs):

install.initialize_options(self, *args, **kwargs)

self.no_prompt = None

def finalize_options(self):

install.finalize_options(self)

if self.no_prompt is None:

"""Specialized version of install_lib, which backs up old bin subdirectories."""

def run(self):

# Determine whether the user is still using an old-style schema

schema_type = get_schema_type(self.install_dir)

# Save any existing 'bin' subdirectory:

if os.path.exists(self.install_dir):

bin_savedir = weeutil.weeutil.move_with_timestamp(self.install_dir)

print "Saved bin subdirectory as %s" % bin_savedir

else:

bin_savedir = None

# Run the superclass's version. This will install all incoming files.

install_lib.run(self)

# If the bin subdirectory previously existed, and if it included

# a 'user' subsubdirectory, then restore it

if bin_savedir:

user_backupdir = os.path.join(bin_savedir, 'user')

if os.path.exists(user_backupdir):

user_dir = os.path.join(self.install_dir, 'user')

distutils.dir_util.copy_tree(user_backupdir, user_dir)

# But, there is one exception: if the old user subdirectory included an

# old-style schema, then it should be overwritten with the new version.

if schema_type == 'old':

incoming_schema_path = os.path.join(bin_dir, 'user/schemas.py')

target_path = os.path.join(self.install_dir, 'user/schemas.py')

"""Specialized version of install_data. Mostly, it deals with upgrading

and merging any old weewx.conf configuration files."""

def initialize_options(self):

# Initialize my superclass's options:

install_data.initialize_options(self)

# Set to None so we inherit whatever setting comes from weewx_install:

self.no_prompt = None

def finalize_options(self):

# Finalize my superclass's options:

install_data.finalize_options(self)

# This will set no_prompt to whatever is in weewx_install:

self.set_undefined_options('install', ('no_prompt', 'no_prompt'))

def copy_file(self, f, install_dir, **kwargs):

# If this is the configuration file, then merge it instead

# of copying it

if f == 'weewx.conf':

rv = self.process_config_file(f, install_dir, **kwargs)

elif f in start_scripts:

rv = self.massage_start_file(f, install_dir, **kwargs)

else:

rv = install_data.copy_file(self, f, install_dir, **kwargs)

return rv

def run(self):

# If there is an existing skins subdirectory, do not overwrite it.

if os.path.exists(os.path.join(self.install_dir, 'skins')):

# Do this by filtering it out of the list of subdirectories to

# be installed:

self.data_files = filter(lambda dat : not dat[0].startswith('skins/'), self.data_files)

remove_obsolete_files(self.install_dir)

# Run the superclass's run():

install_data.run(self)

def process_config_file(self, f, install_dir, **kwargs):

global stn_info

# Open up and parse the distribution config file:

try:

dist_config_dict = configobj.ConfigObj(f, file_error=True)

except IOError, e:

sys.exit(str(e))

except SyntaxError, e:

sys.exit("Syntax error in distribution configuration file '%s': %s"

% (f, e))

# The path where the weewx.conf configuration file will be installed

install_path = os.path.join(install_dir, os.path.basename(f))

# Do we have an old config file?

if os.path.isfile(install_path):

# Yes. Read it

config_path, config_dict = weecfg.read_config(install_path, None)

if DEBUG:

print "Old configuration file found at", config_path

# Update the old configuration file to the current version,

# then merge it into the distribution file

weecfg.update_and_merge(config_dict, dist_config_dict)

else:

# No old config file. Use the distribution file, then, if we can,

# prompt the user for station specific info

config_dict = dist_config_dict

if not self.no_prompt:

# Prompt the user for the station information:

stn_info = weecfg.prompt_for_info()

driver = weecfg.prompt_for_driver(stn_info.get('driver'))

stn_info['driver'] = driver

stn_info.update(weecfg.prompt_for_driver_settings(driver))

if DEBUG:

print "Station info =", stn_info

weecfg.modify_config(config_dict, stn_info, DEBUG)

# Set the WEEWX_ROOT

config_dict['WEEWX_ROOT'] = os.path.normpath(install_dir)

# Finally, reorder it to the canonical form

weecfg.reorder_to_ref(config_dict)

# Time to write it out. Get a temporary file:

with tempfile.NamedTemporaryFile("w") as tmpfile:

# Write the finished configuration file to it:

config_dict.write(tmpfile)

tmpfile.flush()

# Save the old config file if it exists:

if not self.dry_run and os.path.exists(install_path):

backup_path = weeutil.weeutil.move_with_timestamp(install_path)

print "Saved old configuration file as %s" % backup_path

# Now install the temporary file (holding the merged config data)

# into the proper place:

rv = install_data.copy_file(self, tmpfile.name, install_path, **kwargs)

# Set the permission bits unless this is a dry run:

if not self.dry_run:

shutil.copymode(f, install_path)

return rv

def massage_start_file(self, f, install_dir, **kwargs):

outname = os.path.join(install_dir, os.path.basename(f))

sre = re.compile(r"WEEWX_ROOT\s*=")

with open(f, 'r') as infile:

with tempfile.NamedTemporaryFile("w") as tmpfile:

for line in infile:

if sre.match(line):

tmpfile.writelines("WEEWX_ROOT=%s\n" % self.install_dir)

else:

tmpfile.writelines(line)

tmpfile.flush()

rv = install_data.copy_file(self, tmpfile.name, outname, **kwargs)

# Set the permission bits unless this is a dry run:

if not self.dry_run:

shutil.copymode(f, outname)

def run(self):

# Run the superclass's version:

install_scripts.run(self)

try:

# Put in a symbolic link for weewxd.py

os.symlink('./weewxd', os.path.join(self.install_dir, 'weewxd.py'))

except OSError:

"""Specialized version of sdist which checks for password information in

the configuration file before creating the distribution.

For other sdist methods, see:

http://epydoc.sourceforge.net/stdlib/distutils.command.sdist.sdist-class.html

"""

def copy_file(self, f, install_dir, **kwargs):

"""Specialized version of copy_file that checks for stray passwords."""

# If this is the configuration file, then check it for passwords

if f == 'weewx.conf':

import configobj

config = configobj.ConfigObj(f)

try:

password = config['StdReport']['FTP']['password']

sys.exit("\n*** FTP password found in configuration file. Aborting ***\n\n")

except KeyError:

pass

try:

password = config['StdRESTful']['Wunderground']['password']

if password != 'replace_me':

sys.exit("\n*** Wunderground password found in configuration file. Aborting ***\n\n")

except KeyError:

pass

try:

password = config['StdRESTful']['PWSweather']['password']

if password != 'replace_me':

sys.exit("\n*** PWSweather password found in configuration file. Aborting ***\n\n")

except KeyError:

pass

# Pass on to my superclass:

"""Remove no longer needed files from the installation

directory, nominally /home/weewx."""

# If the file #upstream.last exists, delete it, as it is no longer used.

try:

os.remove(os.path.join(install_dir, 'public_html/#upstream.last'))

except OSError:

pass

# If the file $WEEWX_INSTALL/readme.htm exists, delete it. It's

# the old readme (since replaced with README)

try:

os.remove(os.path.join(install_dir, 'readme.htm'))

except OSError:

pass

# If the file $WEEWX_INSTALL/CHANGES.txt exists, delete it. It's

# been moved to the docs subdirectory and renamed

try:

os.remove(os.path.join(install_dir, 'CHANGES.txt'))

except OSError:

pass

# The directory start_scripts is no longer used

shutil.rmtree(os.path.join(install_dir, 'start_scripts'), True)

# The file docs/README.txt is now gone

try:

os.remove(os.path.join(install_dir, 'docs/README.txt'))

except OSError:

pass

# If the file docs/CHANGES.txt exists, delete it. It's been renamed

# to docs/changes.txt

try:

os.remove(os.path.join(install_dir, 'docs/CHANGES.txt'))

except OSError:

pass

# setup.py is no longer left in WEEWX_ROOT.

try:

os.remove(os.path.join(install_dir, 'setup.py'))

except OSError:

"""Checks whether the schema in user.schemas is a new style or old style

schema.

bin_dir: The directory to be checked. This is nominally /home/weewx/bin.

Returns:

'none': There is no schema at all.

'old' : It is an old-style schema.

'new' : It is a new-style schema

"""

tmp_path = list(sys.path)

sys.path.insert(0, bin_dir)

try:

import user.schemas

except ImportError:

# There is no existing schema at all.

result = 'none'

else:

# There is a schema. Determine if it is old-style or new-style

try:

# Try the old style 'drop_list'. If it fails, it must be

# a new-style schema

_ = user.schemas.drop_list # @UnusedVariable @UndefinedVariable

except AttributeError:

# New style schema

result = 'new'

else:

# It did not fail. Must be an old-style schema

result = 'old'

finally:

del user.schemas

# Restore the path

sys.path = tmp_path