From 497dc751840a109e561cbe932599d847a43e1e24 Mon Sep 17 00:00:00 2001
From: finler <finler@gmail.com>
Date: Fri, 3 Jun 2022 09:20:08 +0800
Subject: [PATCH] Update retrieve_cassandraDB.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

有的省份swan数据使用bz2压缩，需要先解压
---
 nmc_met_io/retrieve_cassandraDB.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/nmc_met_io/retrieve_cassandraDB.py b/nmc_met_io/retrieve_cassandraDB.py
index 9210b5a..4c69349 100644
--- a/nmc_met_io/retrieve_cassandraDB.py
+++ b/nmc_met_io/retrieve_cassandraDB.py
@@ -1626,7 +1626,8 @@ def get_swan_radar(directory, filename=None, suffix="*.000", scale=[0.1, 0],
             if byteArray == '':
                 print('There is no data ' + filename + ' in ' + directory)
                 return None
-
+            if filename.endswith('.BZ2') or filename.endswith('.bz2'):
+                byteArray = bz2.decompress(byteArray)
             # define head structure
             head_dtype = [
                 ('ZonName', 'S12'),