Coverage for src/manydataapi/parsers/dataframe_helper.py: 100%

1""" 

2@file 

3@brief Helpers about dataframe. 

4""" 

5import os 

6 

7 

8def dataframe_to(df, out, **kwargs): 

9 """ 

10 Exports a dataframe into the write format. 

11 The function uses the file extension to choose the 

12 right exporter. 

13 

14 @param df dataframe 

15 @param out filename 

16 @param kwargs additional parameter to the exporter 

17 """ 

18 if isinstance(out, str): 

19 ext = os.path.splitext(out)[-1][1:] 

20 if ext == 'xlsx': 

21 ext = 'excel' 

22 meth = "to_" + ext 

23 if hasattr(df, meth): 

24 m = getattr(df, meth) 

25 m(out, **kwargs) 

26 else: 

27 raise RuntimeError("Method '{}' does not exist in type {}.".format( 

28 meth, type(df))) 

29 else: 

30 raise TypeError("out must be a string.")