Changeset 35889

Timestamp:

2022-01-06T09:33:19+13:00 (2 years ago)

Author:

davidb

Message:

Support for command-line start and end years added in

File:

• main/trunk/model-sites-dev/eurovision-lod/collect/eurovision/prepare/08b-COPY-AUDIO-FEATURES-TO-IMPORT.py (modified) (3 diffs)

main/trunk/model-sites-dev/eurovision-lod/collect/eurovision/prepare/08b-COPY-AUDIO-FEATURES-TO-IMPORT.py

@@ -1 +1 @@
 #!/usr/bin/env python3
+
+import argparse
 
 import os
@@ -11 +13 @@
 
 
+parser = argparse.ArgumentParser(description="Copy Essentia generated audio features into Greenstone3 collection's '../import' directory")
+parser.add_argument('--startyear', type=int, default=None,
+                    help='Start year range of the Eurovision Song Contest')
+parser.add_argument('--endyear', type=int, default=None,
+                    help='End year range of the Eurovision Song Contest')
+args = parser.parse_args()
+
+start_year = int(args.startyear) if args.startyear is not None  else None
+end_year   = int(args.endyear)   if args.endyear   is not None  else None
+
+#if args.endyear is not None:
+#     end_year = int(args.sendyear)
+     
 years = [d for d in os.listdir(audio_dir) if os.path.isdir(os.path.join(audio_dir, d))]
 
+sorted_years = sorted(years)
+
+opt_filtered_sorted_years = []
+
+for y in sorted_years:
+     
+     if (start_year is not None) and int(y) < start_year:
+          continue
+     
+     if (end_year is not None) and int(y) > end_year:
+          continue
+
+     opt_filtered_sorted_years.append(y)
+     
 
 print("Reading in Audio-Feature JSON files by year")
@@ -19 +48 @@
 
 
-for year in sorted(years):
+for year in opt_filtered_sorted_years:
 #     print("  " + year)