onetimedownload.py 2.38 KB
Newer Older
Nicolas Delaby's avatar
Nicolas Delaby committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
##############################################################################
#
# Copyright (c) 2010 Vifib SARL and Contributors. All Rights Reserved.
#
# WARNING: This program as such is intended to be used by professional
# programmers who take the whole responsibility of assessing all potential
# consequences resulting from its eventual inadequacies and bugs
# End users who are looking for a ready-to-use solution with commercial
# guarantees and support are strongly adviced to contract a Free Software
# Service Company
#
# This program is Free Software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 3
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
#
##############################################################################
import os
import sys
from optparse import OptionParser, Option
30
from six.moves.urllib.request import urlopen
Nicolas Delaby's avatar
Nicolas Delaby committed
31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57


class Parser(OptionParser):
  """
  Parse all arguments.
  """

  def check_args(self):
    """
    Check arguments
    """
    (options, args) = self.parse_args()
    if len(args) != 2:
      self.error("Incorrect number of arguments")
    url, file_path = args

    return url, file_path

class Config:
  def setConfig(self, url, file_path):
    """
    Set options given by parameters.
    """
    self.url = url
    self.file_path = file_path

def onetimedownload(url, file_path):
58
  url_file = urlopen(url)
Nicolas Delaby's avatar
Nicolas Delaby committed
59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74
  data = url_file.read()
  file_object = open(file_path, 'w')
  file_object.write(data)
  file_object.close()

def main():
  "Run default configuration."
  usage = "usage: onetimedownload URL FILE_PATH"

  try:
    # Parse arguments
    config = Config()
    config.setConfig(*Parser(usage=usage).check_args())

    onetimedownload(config.url, config.file_path)
    return_code = 0
75
  except SystemExit as err:
Nicolas Delaby's avatar
Nicolas Delaby committed
76 77 78 79 80
    # Catch exception raise by optparse
    return_code = err

  sys.exit(return_code)