from datetime import datetime
import smarttypes, sys, string
from smarttypes.config import *
from smarttypes.model.twitter_credentials import TwitterCredentials
from smarttypes.model.twitter_user import TwitterUser
from smarttypes.model.twitter_reduction import TwitterReduction
from smarttypes.utils.postgres_handle import PostgresHandle
postgres_handle = PostgresHandle(smarttypes.connection_string)
if __name__ == "__main__":
user_count_list = TwitterReduction.get_user_reduction_counts(
postgres_handle)
for user, count in user_count_list:
if count == 1:
creds = TwitterCredentials.get_by_twitter_id(
user.id, postgres_handle)
creds.maybe_send_initial_map_email()
def credentials(self):
from smarttypes.model.twitter_credentials import TwitterCredentials
creds = TwitterCredentials.get_by_twitter_id(self.id, self.postgres_handle)
if not creds:
creds = TwitterCredentials.get_by_root_user_id(self.id, self.postgres_handle)
return creds
def credentials(self):
from smarttypes.model.twitter_credentials import TwitterCredentials
creds = TwitterCredentials.get_by_twitter_id(self.id, self.postgres_handle)
if not creds:
creds = TwitterCredentials.get_by_root_user_id(self.id, self.postgres_handle)
return creds
string.ljust(root_username, 20),
string.ljust(creds.email if creds.email else '', 30)
)
if __name__ == "__main__":
"""
if no args, show all creds
if args, first arg is creds_username, second is root_username
"""
if len(sys.argv) == 1:
list_cred_details()
elif len(sys.argv) == 2:
creds_user = TwitterUser.by_screen_name(sys.argv[1], postgres_handle)
creds = TwitterCredentials.get_by_twitter_id(creds_user.id, postgres_handle)
creds.root_user_id = None
creds.save()
postgres_handle.connection.commit()
else:
creds_user = TwitterUser.by_screen_name(sys.argv[1], postgres_handle)
root_user = TwitterUser.by_screen_name(sys.argv[2], postgres_handle)
if not root_user:
api_user = creds_user.credentials.api_handle.get_user(screen_name=sys.argv[2])
root_user = TwitterUser.upsert_from_api_user(api_user, postgres_handle)
postgres_handle.connection.commit()
creds = TwitterCredentials.get_by_twitter_id(creds_user.id, postgres_handle)
creds.root_user_id = root_user.id
creds.save()
postgres_handle.connection.commit()
