Update projects/Scraping Medium Articles/scraping_medium.py

Co-authored-by: Ravishankar Chavare <[email protected]>

Author: namanshah01

projects/Scraping Medium Articles/scraping_medium.py

@@ -11,7 +11,7 @@
 def get_page():
 	url = input('Enter url of a medium article: ')
 	# handling possible error
-	if (url[:18] != 'https://medium.com/') and (url[:17] != 'http://medium.com'):
+	if not re.match(r'https?://medium.com/',url):
 		print('Please enter a valid website, or make sure it is a medium article')
 		sys.exit(1)
 	res = requests.get(url)
@@ -63,4 +63,4 @@ def save_file(fin):
 if __name__ == '__main__':
 	url, soup = get_page()
 	fin = collect_text(url, soup)
-	save_file(fin)
+	save_file(fin)