Skip to main content

Add another test case for fluser_lookup

ID
b9c5a67
date
2024-03-07 13:46:10+00:00
author
Alex Chan <alex@alexwlchan.net>
parent
d2183c0
message
Add another test case for fluser_lookup
changed files
2 files, 8 additions

Changed files

flickr/fluser_lookup.py (1553) → flickr/fluser_lookup.py (1748)

diff --git a/flickr/fluser_lookup.py b/flickr/fluser_lookup.py
index 08749a2..2c36307 100755
--- a/flickr/fluser_lookup.py
+++ b/flickr/fluser_lookup.py
@@ -30,6 +30,10 @@ def get_user_id(user_text: str) -> str:
     ):
         return {"path_alias": u.path[1]}
 
+    # e.g. https://www.flickr.com/photos/powerhouse_museum/2532449275/
+    if u.host == "www.flickr.com" and len(u.path) >= 2 and u.path[0] == "photos":
+        return {"path_alias": u.path[1]}
+
 
 if __name__ == "__main__":
     try:

flickr/test_fluser_lookup.py (356) → flickr/test_fluser_lookup.py (501)

diff --git a/flickr/test_fluser_lookup.py b/flickr/test_fluser_lookup.py
index 2b2835f..f6c9a05 100644
--- a/flickr/test_fluser_lookup.py
+++ b/flickr/test_fluser_lookup.py
@@ -10,6 +10,10 @@ from fluser_lookup import get_user_id
             "https://commons.flickr.org/members/csj_canada_archives/",
             {"path_alias": "csj_canada_archives"},
         ),
+        (
+            "https://www.flickr.com/photos/powerhouse_museum/2532449275/",
+            {"path_alias": "powerhouse_museum"},
+        ),
     ],
 )
 def test_get_user_id(user_text, user_id):