SFR_1826_NYPLLoginFlag

CHANGELOG.md

@@ -1,5 +1,10 @@
 # CHANGELOG
 
+## unreleased version -- v0.12.4
+## Added
+- New script to add nypl_login flag to Links objects
+## Fixed
+
 ## 2023-09-05 version -- v0.12.3
 ## Removed
 - The "*" character is escaped from queries passed to ElasticSearch, limiting wildcard searches

scripts/__init__.py

@@ -11,4 +11,5 @@
 from .interstitialPagesSFR1410 import main as interstitialPages
 from .ingestS3Metadata import main as ingestS3
 from .updatePubLocationAndLinks import main as updateLocationAndLinks
-from .countCABooks import main as countCABooks
+from .countCABooks import main as countCA
+from .nyplLoginFlags import main as nyplFlags

scripts/nyplLoginFlags.py

@@ -0,0 +1,35 @@
+import os
+
+from model import Link
+from managers import DBManager
+from sqlalchemy import or_
+import json
+
+def main():
+
+    '''Updating NYPL Link flags with a new nypl_login flag'''
+
+    dbManager = DBManager(
+        user= os.environ.get('POSTGRES_USER', None),
+        pswd= os.environ.get('POSTGRES_PSWD', None),
+        host= os.environ.get('POSTGRES_HOST', None),
+        port= os.environ.get('POSTGRES_PORT', None),
+        db= os.environ.get('POSTGRES_NAME', None)
+    )
+
+    dbManager.generateEngine()
+
+    dbManager.createSession()
+
+    for link in dbManager.session.query(Link) \
+        .filter(or_(Link.media_type == 'application/html+edd', Link.media_type == 'application/x.html+edd')).all():
+            if link.flags['edd'] == True:
+                #The link.flags doesn't update if the dict method isn't called on it
+                newLinkFlag = dict(link.flags)
+                newLinkFlag['nypl_login'] = True
+                link.flags = newLinkFlag
+
+    dbManager.commitChanges()
+
+if __name__ == '__main__':
+    main()