1 #! /usr/bin/python2.5

     2 

     3 # Copyright 2009 the Melange authors.

     4 #

     5 # Licensed under the Apache License, Version 2.0 (the "License");

     6 # you may not use this file except in compliance with the License.

     7 # You may obtain a copy of the License at

     8 #

     9 #   http://www.apache.org/licenses/LICENSE-2.0

    10 #

    11 # Unless required by applicable law or agreed to in writing, software

    12 # distributed under the License is distributed on an "AS IS" BASIS,

    13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

    14 # See the License for the specific language governing permissions and

    15 # limitations under the License.

    16 

    17 """A script to generate the app.yaml from the template with an application

    18 name filled in.

    19 

    20 Usage:

    21   gen_app_yaml.py [-f] APPLICATION_NAME

    22 

    23 Arguments:

    24   APPLICATION_NAME: the name to use for the application (no underscores)

    25 

    26 Flags:

    27   -f:  overwrite an existing app.yaml (default=false)

    28 """

    29 

    30 from __future__ import with_statement

    31 

    32 __authors__ = [

    33     # alphabetical order by last name, please

    34     '"Dan Bentley" <dbentley@google.com>',

    35     ]

    36 

    37 import os

    38 import sys

    39 

    40 

    41 def generateAppYaml(application_name, force=False):

    42   """Generate the app.yaml file.

    43 

    44   Args:

    45     application_name: str, the name to write into the application filed

    46     force: bool, whether to overwrite an existing app.yaml

    47   """

    48   scripts_directory = os.path.dirname(__file__)

    49   app_dir = os.path.abspath(os.path.join(scripts_directory, '../app'))

    50   template_path = os.path.join(app_dir, 'app.yaml.template')

    51   app_yaml_path = os.path.join(app_dir, 'app.yaml')

    52   if not os.path.exists(template_path):

    53     sys.exit("Template file %s non-existent. Corrupt client?" % template_path)

    54   if os.path.exists(app_yaml_path):

    55     if not force:

    56       sys.exit("%s exists; exiting. To overwrite, pass -f on the command-line"

    57                % app_yaml_path)

    58   with open(template_path) as infile:

    59     template_contents = infile.read()

    60   app_yaml_contents = template_contents.replace(

    61       '# application: FIXME',

    62       'application: '+ application_name)

    63   with open(app_yaml_path, 'w') as outfile:

    64     outfile.write(app_yaml_contents)

    65   # TODO(dbentley): should this be done via logging?

    66   print "Wrote application name %s to %s." % (application_name, app_yaml_path)

    67 

    68 

    69 def usage(msg):

    70   """Print an error message and the usage of the program; then quit.

    71   """

    72   sys.exit('Error: %s\n\n%s' % (msg, __doc__))

    73 

    74 

    75 def main(args):

    76   args = args[1:] # strip off the binary name

    77   if not args:

    78     usage("No arguments supplied.")

    79   if args[0] == '-f':

    80     force = True

    81     args = args[1:]

    82   else:

    83     force = False

    84   if len(args) != 1:

    85     usage("No application name supplied.")

    86   application_name = args[0]

    87   generateAppYaml(application_name, force=force)

    88 

    89 

    90 if __name__ == '__main__':

    91   main(sys.argv)