From c172937fcba06200158cd7065a3a6d22593d75c4 Mon Sep 17 00:00:00 2001
From: aisuneko <iceneko@protonmail.ch>
Date: Fri, 2 Jul 2021 23:23:07 +0800
Subject: [PATCH] implement custom useragent flag

---
 ncm/start.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/ncm/start.py b/ncm/start.py
index b5240e8..dc115db 100644
--- a/ncm/start.py
+++ b/ncm/start.py
@@ -9,10 +9,11 @@ from ncm.downloader import get_song_info_by_id
 from ncm.downloader import download_song_by_id
 from ncm.downloader import download_song_by_song
 from ncm.downloader import format_string
+from ncm.constants import headers
 
 # load the config first
 config.load_config()
-api = CloudApi()
+#api = CloudApi()
 
 
 def download_hot_songs(artist_id):
@@ -64,7 +65,12 @@ def main():
                         help='Download an album all songs by album_id')
     parser.add_argument('-p', metavar='playlist_id', dest='playlist_id',
                         help='Download a playlist all songs by playlist_id')
+    parser.add_argument('-ua', metavar='user_agent', dest='user_agent',
+                        help='Specify the User-Agent to be used when downloading')
     args = parser.parse_args()
+    if args.user_agent:
+        headers.update({'User-Agent':args.user_agent})
+        #api = CloudApi()
     if args.song_id:
         download_song_by_id(get_parse_id(args.song_id), config.DOWNLOAD_DIR)
     elif args.song_ids:
@@ -80,3 +86,4 @@ def main():
 
 if __name__ == '__main__':
     main()
+    
\ No newline at end of file