From 5f091518abe7e2b3ff1ae9352ba5080bc4eccf2d Mon Sep 17 00:00:00 2001 From: Seo Sanghyeon Date: Sun, 14 May 2023 11:53:46 +0900 Subject: [PATCH] Use __main__ --- lobsters/search.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/lobsters/search.py b/lobsters/search.py index 82327f8..456f152 100644 --- a/lobsters/search.py +++ b/lobsters/search.py @@ -1,3 +1,5 @@ +import argparse + import lxml.html import requests @@ -12,9 +14,9 @@ def search(topic): text = item.text_content() print(f' * {text}') -import argparse -parser = argparse.ArgumentParser() -parser.add_argument('topic') -args = parser.parse_args() - -search(args.topic) +if __name__ == '__main__': + parser = argparse.ArgumentParser() + parser.add_argument('topic') + args = parser.parse_args() + topic = args.topic + search(topic)