diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index c849700..bb57cfa 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -1,11 +1,11 @@ repos: - repo: https://github.com/psf/black - rev: 23.3.0 + rev: 25.1.0 hooks: - id: black - repo: https://github.com/pycqa/isort - rev: 5.12.0 + rev: 6.0.1 hooks: - id: isort args: ["--profile", "black"] \ No newline at end of file diff --git a/sample.py b/sample.py index 9983c1e..7b02be8 100644 --- a/sample.py +++ b/sample.py @@ -2,15 +2,18 @@ # -*- coding: utf-8 -*- # vim:fenc=utf-8 -from src.bots.simple_moving_average import TradeBotSimpleMovingAverage -from src.bots.config import TradingConfig, RiskManagement -import time -import sys -from datetime import datetime, time as dt_time -import logging import json +import logging +import sys +import time +from datetime import datetime +from datetime import time as dt_time + import pandas as pd +from src.bots.config import RiskManagement, TradingConfig +from src.bots.simple_moving_average import TradeBotSimpleMovingAverage + logging.basicConfig( level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s", diff --git a/src/bots/base_trade_bot.py b/src/bots/base_trade_bot.py index 667cb58..7a51fde 100644 --- a/src/bots/base_trade_bot.py +++ b/src/bots/base_trade_bot.py @@ -3,15 +3,17 @@ # vim:fenc=utf-8 import logging -from typing import Dict, Optional, List +from dataclasses import dataclass +from datetime import datetime, timezone +from typing import Dict, List, Optional + +import numpy as np import pandas as pd import pyotp import robin_stocks.robinhood as robinhood -from datetime import datetime, timezone -import numpy as np -from dataclasses import dataclass + +from src.bots.config import OrderType, StrategyType, TradingConfig from src.utilities import RobinhoodCredentials -from src.bots.config import TradingConfig, StrategyType, OrderType # Configure logging to write to a file logging.basicConfig( diff --git a/src/bots/config.py b/src/bots/config.py index 4675a51..e254055 100644 --- a/src/bots/config.py +++ b/src/bots/config.py @@ -3,10 +3,11 @@ # vim:fenc=utf-8 from dataclasses import dataclass, field -from typing import Dict, List +from datetime import timedelta from enum import Enum, auto +from typing import Dict, List + import numpy as np -from datetime import timedelta class OrderType(Enum): diff --git a/src/bots/simple_moving_average.py b/src/bots/simple_moving_average.py index e63de1b..a0ca070 100644 --- a/src/bots/simple_moving_average.py +++ b/src/bots/simple_moving_average.py @@ -2,13 +2,15 @@ # -*- coding: utf-8 -*- # vim:fenc=utf-8 -from src.bots.base_trade_bot import OrderType, TradeBot, OrderResult -from src.bots.config import TradingConfig, StrategyType -import pandas as pd -import numpy as np -from typing import Optional, Dict, Tuple import logging from datetime import datetime, timezone +from typing import Dict, Optional, Tuple + +import numpy as np +import pandas as pd + +from src.bots.base_trade_bot import OrderResult, OrderType, TradeBot +from src.bots.config import StrategyType, TradingConfig logger = logging.getLogger(__name__) diff --git a/src/bots/twitter_sentiments.py b/src/bots/twitter_sentiments.py index 6edf7e2..6d596d6 100644 --- a/src/bots/twitter_sentiments.py +++ b/src/bots/twitter_sentiments.py @@ -3,17 +3,19 @@ # vim:fenc=utf-8 import logging +import re +from collections import defaultdict from dataclasses import dataclass from datetime import datetime -from typing import List, Optional, Dict, Any, Tuple +from typing import Any, Dict, List, Optional, Tuple + import numpy as np import tweepy -from utilities import TwitterCredentials -from vaderSentiment.vaderSentiment import SentimentIntensityAnalyzer from textblob import TextBlob -import re -from collections import defaultdict +from vaderSentiment.vaderSentiment import SentimentIntensityAnalyzer + from src.bots.config import OrderType +from utilities import TwitterCredentials @dataclass diff --git a/src/bots/volume_weighted_average_price.py b/src/bots/volume_weighted_average_price.py index 9b08921..50669cb 100644 --- a/src/bots/volume_weighted_average_price.py +++ b/src/bots/volume_weighted_average_price.py @@ -4,13 +4,15 @@ import logging from dataclasses import dataclass -import pandas as pd -import numpy as np -from typing import Optional, Dict, Any, List from datetime import datetime -from src.bots.config import OrderType +from typing import Any, Dict, List, Optional + +import numpy as np +import pandas as pd import robin_stocks.robinhood as robinhood +from src.bots.config import OrderType + @dataclass class VWAPMetrics: