move trades_get_pairs to parent class
This commit is contained in:
parent
a06eee300a
commit
8116ca847b
@ -1,7 +1,5 @@
|
|||||||
import logging
|
import logging
|
||||||
import re
|
from typing import Optional
|
||||||
from pathlib import Path
|
|
||||||
from typing import List, Optional
|
|
||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import pandas as pd
|
import pandas as pd
|
||||||
@ -101,18 +99,6 @@ class HDF5DataHandler(IDataHandler):
|
|||||||
"""
|
"""
|
||||||
raise NotImplementedError()
|
raise NotImplementedError()
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def trades_get_pairs(cls, datadir: Path) -> List[str]:
|
|
||||||
"""
|
|
||||||
Returns a list of all pairs for which trade data is available in this
|
|
||||||
:param datadir: Directory to search for ohlcv files
|
|
||||||
:return: List of Pairs
|
|
||||||
"""
|
|
||||||
_tmp = [re.search(r'^(\S+)(?=\-trades.h5)', p.name)
|
|
||||||
for p in datadir.glob("*trades.h5")]
|
|
||||||
# Check if regex found something and only return these results to avoid exceptions.
|
|
||||||
return [cls.rebuild_pair_from_filename(match[0]) for match in _tmp if match]
|
|
||||||
|
|
||||||
def trades_store(self, pair: str, data: TradeList) -> None:
|
def trades_store(self, pair: str, data: TradeList) -> None:
|
||||||
"""
|
"""
|
||||||
Store trades data (list of Dicts) to file
|
Store trades data (list of Dicts) to file
|
||||||
|
@ -152,13 +152,17 @@ class IDataHandler(ABC):
|
|||||||
"""
|
"""
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
@abstractmethod
|
|
||||||
def trades_get_pairs(cls, datadir: Path) -> List[str]:
|
def trades_get_pairs(cls, datadir: Path) -> List[str]:
|
||||||
"""
|
"""
|
||||||
Returns a list of all pairs for which trade data is available in this
|
Returns a list of all pairs for which trade data is available in this
|
||||||
:param datadir: Directory to search for ohlcv files
|
:param datadir: Directory to search for ohlcv files
|
||||||
:return: List of Pairs
|
:return: List of Pairs
|
||||||
"""
|
"""
|
||||||
|
_ext = cls._get_file_extension()
|
||||||
|
_tmp = [re.search(r'^(\S+)(?=\-trades.' + _ext + ')', p.name)
|
||||||
|
for p in datadir.glob(f"*trades.{_ext}")]
|
||||||
|
# Check if regex found something and only return these results to avoid exceptions.
|
||||||
|
return [cls.rebuild_pair_from_filename(match[0]) for match in _tmp if match]
|
||||||
|
|
||||||
@abstractmethod
|
@abstractmethod
|
||||||
def trades_store(self, pair: str, data: TradeList) -> None:
|
def trades_store(self, pair: str, data: TradeList) -> None:
|
||||||
|
@ -1,7 +1,5 @@
|
|||||||
import logging
|
import logging
|
||||||
import re
|
from typing import Optional
|
||||||
from pathlib import Path
|
|
||||||
from typing import List, Optional
|
|
||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
from pandas import DataFrame, read_json, to_datetime
|
from pandas import DataFrame, read_json, to_datetime
|
||||||
@ -99,18 +97,6 @@ class JsonDataHandler(IDataHandler):
|
|||||||
"""
|
"""
|
||||||
raise NotImplementedError()
|
raise NotImplementedError()
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def trades_get_pairs(cls, datadir: Path) -> List[str]:
|
|
||||||
"""
|
|
||||||
Returns a list of all pairs for which trade data is available in this
|
|
||||||
:param datadir: Directory to search for ohlcv files
|
|
||||||
:return: List of Pairs
|
|
||||||
"""
|
|
||||||
_tmp = [re.search(r'^(\S+)(?=\-trades.json)', p.name)
|
|
||||||
for p in datadir.glob(f"*trades.{cls._get_file_extension()}")]
|
|
||||||
# Check if regex found something and only return these results to avoid exceptions.
|
|
||||||
return [cls.rebuild_pair_from_filename(match[0]) for match in _tmp if match]
|
|
||||||
|
|
||||||
def trades_store(self, pair: str, data: TradeList) -> None:
|
def trades_store(self, pair: str, data: TradeList) -> None:
|
||||||
"""
|
"""
|
||||||
Store trades data (list of Dicts) to file
|
Store trades data (list of Dicts) to file
|
||||||
|
Loading…
Reference in New Issue
Block a user