Move trim_dataframe from history to converter
This commit is contained in:
		| @@ -2,10 +2,12 @@ | ||||
| Functions to convert data from one format to another | ||||
| """ | ||||
| import logging | ||||
| from datetime import datetime, timezone | ||||
|  | ||||
| import pandas as pd | ||||
| from pandas import DataFrame, to_datetime | ||||
|  | ||||
| from freqtrade.configuration.timerange import TimeRange | ||||
|  | ||||
| logger = logging.getLogger(__name__) | ||||
|  | ||||
| @@ -112,6 +114,23 @@ def ohlcv_fill_up_missing_data(dataframe: DataFrame, timeframe: str, pair: str) | ||||
|     return df | ||||
|  | ||||
|  | ||||
| def trim_dataframe(df: DataFrame, timerange: TimeRange, df_date_col: str = 'date') -> DataFrame: | ||||
|     """ | ||||
|     Trim dataframe based on given timerange | ||||
|     :param df: Dataframe to trim | ||||
|     :param timerange: timerange (use start and end date if available) | ||||
|     :param: df_date_col: Column in the dataframe to use as Date column | ||||
|     :return: trimmed dataframe | ||||
|     """ | ||||
|     if timerange.starttype == 'date': | ||||
|         start = datetime.fromtimestamp(timerange.startts, tz=timezone.utc) | ||||
|         df = df.loc[df[df_date_col] >= start, :] | ||||
|     if timerange.stoptype == 'date': | ||||
|         stop = datetime.fromtimestamp(timerange.stopts, tz=timezone.utc) | ||||
|         df = df.loc[df[df_date_col] <= stop, :] | ||||
|     return df | ||||
|  | ||||
|  | ||||
| def order_book_to_dataframe(bids: list, asks: list) -> DataFrame: | ||||
|     """ | ||||
|     Gets order book list, returns dataframe with below format per suggested by creslin | ||||
|   | ||||
		Reference in New Issue
	
	Block a user