Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions .pre-commit-config.yaml
Original file line number Diff line number Diff line change
@@ -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"]
15 changes: 9 additions & 6 deletions sample.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down
12 changes: 7 additions & 5 deletions src/bots/base_trade_bot.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down
5 changes: 3 additions & 2 deletions src/bots/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down
12 changes: 7 additions & 5 deletions src/bots/simple_moving_average.py
Original file line number Diff line number Diff line change
Expand Up @@ -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__)

Expand Down
12 changes: 7 additions & 5 deletions src/bots/twitter_sentiments.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
10 changes: 6 additions & 4 deletions src/bots/volume_weighted_average_price.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down