Skip to content

Commit

Permalink
Merge pull request #12 from uclatommy/hotfix-0.2.1
Browse files Browse the repository at this point in the history
Fix connection problem and sentiment calc.
  • Loading branch information
Thomas Chen, ASA committed Feb 26, 2017
2 parents 8051035 + eea2105 commit 4a90d6e
Show file tree
Hide file tree
Showing 3 changed files with 8 additions and 9 deletions.
5 changes: 3 additions & 2 deletions tweetfeels/tweetdata.py
Expand Up @@ -13,10 +13,11 @@ def __init__(self, file='feels.sqlite'):
self.make_feels_db(self._db)
self._debug = False
self.chunksize=1000
self.fields = self._fields

@property
def fields(self):
conn = sqlite3.connect(self._db)
def _fields(self):
conn = sqlite3.connect(self._db, detect_types=sqlite3.PARSE_DECLTYPES)
c = conn.cursor()
c.execute("select * from tweets")
fields=tuple([f[0] for f in c.description])
Expand Down
10 changes: 4 additions & 6 deletions tweetfeels/tweetfeels.py
Expand Up @@ -24,24 +24,21 @@ class TweetFeels(object):
:ivar lang: A list of languages to include in tweet gathering.
"""
_db_factory = (lambda db: TweetData(db))
_auth_factory = (
lambda cred: OAuthHandler(cred[0], cred[1]
).set_access_token(cred[2], cred[3])
)
_listener_factory = (lambda ctrl: TweetListener(ctrl))
_stream_factory = (lambda auth, listener: Stream(auth, listener))

def __init__(self, credentials, tracking=[], db='feels.sqlite'):
self._feels = TweetFeels._db_factory(db)
_auth = TweetFeels._auth_factory(credentials)
_auth = OAuthHandler(credentials[0], credentials[1])
_auth.set_access_token(credentials[2], credentials[3])
self._listener = TweetFeels._listener_factory(self)
self._stream = TweetFeels._stream_factory(_auth, self._listener)
self.tracking = tracking
self.lang = ['en']
self._sentiment = 0
self._filter_level = 'low'
self.calc_every_n = 10
self._latest_calc = None
self._latest_calc = 0
self._tweet_buffer = deque()
self.buffer_limit = 50

Expand Down Expand Up @@ -123,4 +120,5 @@ def avg_sentiment(df):
df = df.sort_index()
for row in df.iteritems():
self._sentiment = self._sentiment*0.99 + row[1]*0.01
self._latest_calc = df.tail(1).index.to_pydatetime()[0]
return self._sentiment
2 changes: 1 addition & 1 deletion tweetfeels/version.py
@@ -1 +1 @@
__version__ = '0.2.0'
__version__ = '0.2.1'

0 comments on commit 4a90d6e

Please sign in to comment.