Merge pull request #5 from umihico/feature/reduce-code

style: reduce code2

Author: umihico

Dockerfile

@@ -1,16 +1,14 @@
 FROM public.ecr.aws/lambda/python:3.8 as build
-RUN mkdir -p /opt/bin/ && \
-    mkdir -p /tmp/downloads && \
-    yum install -y unzip && \
-    curl -SL https://chromedriver.storage.googleapis.com/2.37/chromedriver_linux64.zip > /tmp/downloads/chromedriver.zip && \
-    curl -SL https://github.com/adieuadieu/serverless-chrome/releases/download/v1.0.0-37/stable-headless-chromium-amazonlinux-2017-03.zip > /tmp/downloads/headless-chromium.zip && \
-    unzip /tmp/downloads/chromedriver.zip -d /opt/bin/ && \
-    unzip /tmp/downloads/headless-chromium.zip -d /opt/bin/
+RUN yum install -y unzip && \
+    curl -SL https://chromedriver.storage.googleapis.com/2.37/chromedriver_linux64.zip > /tmp/chromedriver.zip && \
+    curl -SL https://github.com/adieuadieu/serverless-chrome/releases/download/v1.0.0-37/stable-headless-chromium-amazonlinux-2017-03.zip > /tmp/headless-chromium.zip && \
+    unzip /tmp/chromedriver.zip -d /opt/ && \
+    unzip /tmp/headless-chromium.zip -d /opt/
 
 FROM public.ecr.aws/lambda/python:3.8
-RUN mkdir -p /opt/bin && pip install selenium
 RUN yum install -y https://dl.google.com/linux/direct/google-chrome-stable_current_x86_64.rpm
-COPY --from=build /opt/bin/headless-chromium /opt/bin/
-COPY --from=build /opt/bin/chromedriver /opt/bin/
+RUN pip install selenium
+COPY --from=build /opt/headless-chromium /opt/
+COPY --from=build /opt/chromedriver /opt/
 COPY test.py ./
 CMD [ "test.handler" ]

test.py

@@ -3,14 +3,14 @@
 
 def handler(event=None, context=None):
     options = webdriver.ChromeOptions()
-    options.binary_location = "/opt/bin/headless-chromium"
+    options.binary_location = "/opt/headless-chromium"
     options.add_argument('--headless')
     options.add_argument('--no-sandbox')
     options.add_argument("--disable-gpu")
     options.add_argument("--window-size=1280x1696")
     options.add_argument("--single-process")
     options.add_argument("--disable-dev-shm-usage")
-    chrome = webdriver.Chrome("/opt/bin/chromedriver",
+    chrome = webdriver.Chrome("/opt/chromedriver",
                               options=options)
     chrome.get("https://umihi.co/")
     return chrome.find_element_by_xpath("//html").text