Bedre støtte for innlogget innhold

Author: Lanjelin

README.md

@@ -32,6 +32,23 @@ Følgende er støttet:
  - Musikkmanuskripter (digimanus)
  - Plakater (digifoto)
  - Programrapport (digiprogramrapport)
+
+<details>
+  <summary>Innlogget innhold</summary>
+  Som innlogget, åpne boka der en kan lese den, åpne Utviklerverktøy og finn Nettverkfanen.  
+  Refresh siden, og nettverksfanen populeres med innhold.  
+  Finn og velg manifest?fields=etcetc i listen, og bla igjen ned og finn Request Headers.  
+
+  Her kopieres innholdet fra authorization og cookie og lagres i en textfil ved scriptet.  
+  Formaten på tekstfilen er
+  ```
+  authorization=4JjcVi6faGF-GhD6wMoXZ80rUkg.*AAJTSQACMDIAAlNLABxxRandomRandomxxSNVpvUTlQxxRandomRandomxxDVFMAAlMxAAIwMQ..
+  cookie=_ga=GA1.1.1234543217.123454321; _hjSessionUser_123454321=eyJpZCI6IjUzOTZmxxRandomRandomxxy1hNDEwLTc0ZjA4NTJhxxRandomRandomxxOjE3MjYwNTEyNzcxxRandomRandomxxW5nIjp0cnVlfQ==;osvosvetc
+  ```
+  Scriptet kjøres deretter med `--cookie` flagget som peker til filen, feks `python3 nbno.py --id blabla --cookie nbno-cookie.txt`
+
+</details>
+
 ```
 bruk: nbno.py [-h] [--id <ID>] [--cover] [--pdf] [--f2pdf] [--url] [--error] 
               [--v] [--resize <int>] [--start <int>] [--stop <int>]
@@ -51,4 +68,5 @@ valgfrie argumenter:
   --resize <int>  Prosent av originalstørrelse på bilder
   --start <int>   Sidetall å starte på
   --stop <int>    Sidetall å stoppe på
+  --cookie <string>  Sti til fil for autentisering
 ```

nbno.py

@@ -77,6 +77,16 @@ def set_folder_path(self, folder_path):
         self.folder_path = folder_path
         self.find_existing_files()
 
+    def load_cookie(self, file_path):
+        with open(file_path) as f:
+            for line in f:
+                if "=" in line:
+                    key, value = map(str.strip, line.split("=", 1))
+                    if key in ["authorization", "cookie"]:
+                        self.session.headers[key] = value
+        if self.verbose:
+            print(self.session.headers)
+
     def find_existing_files(self):
         filelist = []
         filelist.extend(glob(os.path.join(f"{self.folder_path}*.jpg")))
@@ -409,6 +419,12 @@ def main():
     optional.add_argument(
         "--stop", metavar="<int>", help="Sidetall å stoppe på", default=False
     )
+    optional.add_argument(
+        "--cookie",
+        metavar="<string>",
+        help="Sti til fil for autentisering",
+        default=False
+    )
     parser._action_groups.append(optional)
     args = parser.parse_args()
 
@@ -444,6 +460,12 @@ def main():
             book.set_to_print_errors()
         if args.v:
             book.verbose_print()
+        if args.cookie:
+            if os.path.exists(args.cookie):
+                book.load_cookie(args.cookie)
+            else:
+                print(f"Fil for autentisering ikke funnet: {args.cookie}")
+                exit()
         if args.resize:
             book.set_resize(int(args.resize))
         if args.start: