From 60cf40380a3019651a96a189491bf6f36cf0887d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mike=20F=C3=A4hrmann?= <mike_faehrmann@web.de>
Date: Mon, 22 Jul 2019 22:15:36 +0200
Subject: [PATCH] [vsco] add user extractor (#331)

---
 docs/supportedsites.rst          |   1 +
 gallery_dl/extractor/__init__.py |   1 +
 gallery_dl/extractor/vsco.py     | 111 +++++++++++++++++++++++++++++++
 scripts/supportedsites.py        |   1 +
 4 files changed, 114 insertions(+)
 create mode 100644 gallery_dl/extractor/vsco.py

diff --git a/docs/supportedsites.rst b/docs/supportedsites.rst
index b3299448f4..80631ee715 100644
--- a/docs/supportedsites.rst
+++ b/docs/supportedsites.rst
@@ -107,6 +107,7 @@ The /b/ Archive      https://thebarchive.com/            Threads
 Tsumino              https://www.tsumino.com/            Galleries, Search Results                          Optional
 Tumblr               https://www.tumblr.com/             Images from Users, Likes, Posts, Tag-Searches      Optional (OAuth)
 Twitter              https://twitter.com/                Media Timelines, Timelines, Tweets                 Optional
+VSCO                 https://vsco.co/                    Images from Users
 Wallhaven            https://wallhaven.cc/               individual Images, Search Results
 Warosu               https://warosu.org/                 Threads
 Weibo                https://www.weibo.com/              Images from Users, Images from Statuses
diff --git a/gallery_dl/extractor/__init__.py b/gallery_dl/extractor/__init__.py
index d95c28d84e..a603d0515d 100644
--- a/gallery_dl/extractor/__init__.py
+++ b/gallery_dl/extractor/__init__.py
@@ -96,6 +96,7 @@
     "tumblr",
     "twitter",
     "vanillarock",
+    "vsco",
     "wallhaven",
     "warosu",
     "weibo",
diff --git a/gallery_dl/extractor/vsco.py b/gallery_dl/extractor/vsco.py
new file mode 100644
index 0000000000..38569186ee
--- /dev/null
+++ b/gallery_dl/extractor/vsco.py
@@ -0,0 +1,111 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2019 Mike Fährmann
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License version 2 as
+# published by the Free Software Foundation.
+
+"""Extractors for https://vsco.co/"""
+
+from .common import Extractor, Message
+from .. import text
+import json
+
+
+BASE_PATTERN = r"(?:https?://)?(?:www\.)?vsco\.co/([^/]+)"
+
+
+class VscoExtractor(Extractor):
+    category = "vsco"
+    root = "https://vsco.co"
+    filename_fmt = "{id}.{extension}"
+    archive_fmt = "{id}"
+
+    def __init__(self, match):
+        Extractor.__init__(self, match)
+        self.user = match.group(1).lower()
+
+    def items(self):
+        yield Message.Version, 1
+        yield Message.Directory, {"user": self.user}
+        for img in self.images():
+            url = "https://" + img["responsive_url"]
+            data = text.nameext_from_url(url, {
+                "id"    : img["_id"],
+                "user"  : self.user,
+                "grid"  : img["grid_name"],
+                "meta"  : img.get("image_meta") or {},
+                "tags"  : [tag["text"] for tag in img.get("tags") or ()],
+                "date"  : text.parse_timestamp(img["upload_date"] // 1000),
+                "width" : img["width"],
+                "height": img["height"],
+                "description": img["description"],
+            })
+            yield Message.Url, url, data
+
+    def _pagination(self, url, params, token, extra):
+        yield from extra
+
+        headers = {
+            "Referer"          : "{}/{}".format(self.root, self.user),
+            "Authorization"    : "Bearer " + token,
+            "X-Client-Platform": "web",
+            "X-Client-Build"   : "1",
+        }
+
+        while True:
+            data = self.request(url, headers=headers, params=params).json()
+            if not data.get("media"):
+                return
+            yield from data["media"]
+            params["page"] += 1
+
+
+class VscoUserExtractor(VscoExtractor):
+    subcategory = "user"
+    directory_fmt = ("{category}", "{user}")
+    pattern = BASE_PATTERN + r"/images/"
+    test = ("https://vsco.co/missuri/images/1", {
+        "range": "1-80",
+        "count": 80,
+        "pattern": r"https://im\.vsco\.co/[^/]+/[0-9a-f/]+/vsco\w+\.\w+",
+        "keyword": {
+            "id"    : str,
+            "user"  : "missuri",
+            "grid"  : "anybodyseenmylife",
+            "meta"  : dict,
+            "tags"  : list,
+            "date"  : "type:datetime",
+            "width" : int,
+            "height": int,
+            "description": str,
+        },
+    })
+
+    def images(self):
+        url = "{}/{}/images/1".format(self.root, self.user)
+        page = self.request(url, notfound="user").text
+        data = json.loads(text.extract(page, "__PRELOADED_STATE__ = ", "<")[0])
+        site_id = str(data["sites"]["siteByUsername"][self.user]["site"]["id"])
+        token = data["users"]["currentUser"]["tkn"]
+
+        url = "https://vsco.co/api/2.0/medias"
+        params = {
+            "site_id": site_id,
+            "page"   : 2,
+            "size"   : "30",
+        }
+
+        extra = []
+        medias = data["medias"]["byId"]
+        for mid in data["medias"]["bySiteId"][site_id]["medias"]["1"]:
+            media = medias[mid]["media"]
+            media["_id"] = media["id"]
+            media["grid_name"] = media["gridName"]
+            media["image_meta"] = media["imageMeta"]
+            media["upload_date"] = media["uploadDate"]
+            media["responsive_url"] = media["responsiveUrl"]
+            extra.append(media)
+
+        return self._pagination(url, params, token, extra)
diff --git a/scripts/supportedsites.py b/scripts/supportedsites.py
index 78ee13a809..5045b07720 100755
--- a/scripts/supportedsites.py
+++ b/scripts/supportedsites.py
@@ -72,6 +72,7 @@
     "smugmug"        : "SmugMug",
     "thebarchive"    : "The /b/ Archive",
     "vanillarock"    : "もえぴりあ",
+    "vsco"           : "VSCO",
     "wikiart"        : "WikiArt.org",
     "worldthree"     : "World Three",
     "xhamster"       : "xHamster",