summaryrefslogtreecommitdiff
path: root/baserockimport/exts/rubygems.to_lorry
diff options
context:
space:
mode:
Diffstat (limited to 'baserockimport/exts/rubygems.to_lorry')
-rwxr-xr-xbaserockimport/exts/rubygems.to_lorry5
1 files changed, 3 insertions, 2 deletions
diff --git a/baserockimport/exts/rubygems.to_lorry b/baserockimport/exts/rubygems.to_lorry
index d5f1efa..0413204 100755
--- a/baserockimport/exts/rubygems.to_lorry
+++ b/baserockimport/exts/rubygems.to_lorry
@@ -76,9 +76,10 @@ class RubyGemLorryGenerator(ImportExtension):
"Loaded %i known source URIs from local metadata.", len(self.known_source_uris))
def process_args(self, args):
- if len(args) != 1:
+ if len(args) not in [1, 2]:
raise ImportException(
- 'Please call me with the name of a RubyGem as an argument.')
+ 'Please call me with the name of a RubyGem as an argument'
+ ' and optionally its version number (format: NAME [VERSION])')
gem_name = args[0]