1
0
mirror of https://github.com/obskyr/khinsider.git synced 2025-03-13 22:49:43 +01:00

Only define the main function if run as a script.

This commit is contained in:
obskyr 2014-10-23 23:46:46 +02:00
parent fbd8709d58
commit d83f49f2ed

View File

@ -81,32 +81,32 @@ def search(term):
# --- And now for the execution. ---
def doIt(): # Only in a function to be able to stop after errors, really.
try:
ostName = sys.argv[1]
except IndexError:
print "No soundtrack specified! As the first parameter, use the name the soundtrack uses in its URL."
return
if not os.path.isdir(ostName):
os.mkdir(ostName)
madeDir = True
try:
download(ostName, ostName, verbose=True)
except IndexError:
searchResults = search(ostName)
print "The soundtrack \"" + ostName + "\" does not seem to exist."
if searchResults: # aww yeah we gon' do some searchin'
print
print "These exist, though:"
for name in searchResults:
print name
if madeDir:
os.rmdir(ostName)
return
if __name__ == '__main__':
def doIt(): # Only in a function to be able to stop after errors, really.
try:
ostName = sys.argv[1]
except IndexError:
print "No soundtrack specified! As the first parameter, use the name the soundtrack uses in its URL."
return
if not os.path.isdir(ostName):
os.mkdir(ostName)
madeDir = True
try:
download(ostName, ostName, verbose=True)
except IndexError:
searchResults = search(ostName)
print "The soundtrack \"" + ostName + "\" does not seem to exist."
if searchResults: # aww yeah we gon' do some searchin'
print
print "These exist, though:"
for name in searchResults:
print name
if madeDir:
os.rmdir(ostName)
return
doIt()