Update something

README.md

@@ -10,8 +10,8 @@ An **infinite** pinterest crawler, crawl image by page.
 # Installation
 1. Download requirements
 ```
-git clone https://github.com/mirusu400/Pinterest-crawler
-cd Pinterest-crawler
+git clone https://github.com/mirusu400/Pinterest-infinite-crawler
+cd Pinterest-infinite-crawler
 pip install -r requirements.txt
 ```
 

main.py

@@ -1,5 +1,6 @@
 from selenium import webdriver
 from pinterest import Pinterest
+from sys import exit
 import yaml
 import os.path
 
@@ -8,7 +9,7 @@
     email = ""
     password = ""
     directory = ""
-    currentdir = os.path.dirname(os.path.realpath(__file__))
+    currentdir = os.getcwd()
     print(currentdir + "/chromedriver.exe")
     # Check chromedriver exists
     if not os.path.isfile(currentdir + "/chromedriver.exe"):

pinterest.py

@@ -1,15 +1,17 @@
 from selenium import webdriver
 from bs4 import BeautifulSoup
-import asyncio
 from imagehelper import *
 from time import sleep
+from sys import exit
+import asyncio
+
 class Pinterest():
     def __init__(self, login, pw):
         self.piclist = []
         options = webdriver.ChromeOptions()
-        # options.add_argument('headless')
-        # options.add_argument('window-size=1920x1080')
-        # options.add_argument("disable-gpu")
+        options.add_argument('headless')
+        options.add_argument('window-size=1920x1080')
+        options.add_argument("disable-gpu")
         self.driver = webdriver.Chrome('chromedriver', chrome_options=options)
         try:
             self.driver.get("https://pinterest.com/login")