1baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet#!/usr/bin/env python
2baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet#
3baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet# Copyright 2016 Google Inc.
4baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet#
5baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet# Use of this source code is governed by a BSD-style license that can be
6baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet# found in the LICENSE file.
7baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet
8baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet
9baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet"""Create the asset."""
10baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet
11baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet
12baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenetimport argparse
13baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenetimport common
14baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenetfrom assets import asset_utils
15baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet
16baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet
17baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenetdef create_asset(target_dir):
18baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet  """Create the asset."""
19baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet  # The common case is to add one or more images to the existing set. Therefore,
20baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet  # download the previous version first.
21baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet  asset = asset_utils.Asset(common.ASSET_NAME, asset_utils.MultiStore())
22baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet  asset.download_current_version(target_dir)
23baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet
24baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet  # Allow the user to modify the contents of the target dir.
25baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet  raw_input('Previous SKImage contents have been downloaded. Please make '
26baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet            'your desired changes in the following directory and press enter '
27baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet            'to continue:\n%s' % target_dir)
28baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet
29baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet
30baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenetdef main():
31baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet  parser = argparse.ArgumentParser()
32baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet  parser.add_argument('--target_dir', '-t', required=True)
33baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet  args = parser.parse_args()
34baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet  create_asset(args.target_dir)
35baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet
36baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet
37baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenetif __name__ == '__main__':
38baaea531c249d3a13b0d04eca59b8bd0ee58d76bborenet  main()
39