def available(): user_id = 1 db = MySQLdb.connect("localhost", "root", "root", "reviews") cursor = db.cursor() try: sql = "SELECT status from user where user_id="+ user_id # print sql cursor.execute(sql) results = cursor.fetchall() if results[0] == 'empty': pass elif result[0] == 'uploaded': start(user_id +".csv") except MySQLdb.Error as e: print e db.rollback()
def test_start_true(self): # mock globals to mimic __main__ try: original = globals()['__name__'] globals()['__name__'] = "__main__" self.assertTrue(begin.start()) finally: globals()['__name__'] = original
quota_dict[namespace_name] = int(quota_info['blockSize']) except ECSClientException: # Secure buckets dont provide their size logger.warning( 'Error found in namespace: %s\nException: %s\n skipping', namespace['name'], Exception) continue logger.debug(size_dict) logger.debug(quota_dict) client.authentication.logout() return size_dict, quota_dict # if __name__ == "__main__": if begin.start(): pass def convert_datetime_timezone(dt, tz1, tz2): tz1 = pytz.timezone(tz1) tz2 = pytz.timezone(tz2) sp = dt.split('T') dt = sp[0] + " " + sp[1] dt = datetime.datetime.strptime(dt, "%Y-%m-%d %H:%M:%S") dt = tz1.localize(dt) dt = dt.astimezone(tz2) dt = dt.strftime("%Y-%m-%d %H:%M:%S") return dt
def test_start_false(self): # begin.start() return false outside __main__ self.assertFalse(begin.start())
def test_not_callable(self): target = mock.Mock with self.assertRaises(ValueError): begin.start(Ellipsis)
for word in candidates: word_frequency.setdefault(word, 0) word_frequency[word] += 1 return sorted(word_frequency.items(), key=lambda item: item[1], reverse=True) @begin.subcommand def run(input_file, output_file='output.txt', verbose=True): f = codecs.open(input_file, encoding='utf-8') text = f.read().replace('\n', ' ') f.close() extractor = KeywordExtractor() candidates = extractor.extract_candidate_words(text) keywords = extractor.compute_word_frequency(candidates) f = codecs.open(output_file, 'w', encoding='utf-8') for word in keywords: f.write(u'{} {}\n'.format(word[0], word[1])) if verbose: print word[0], word[1] f.close() @begin.start def main(): pass if begin.start(): pass
from begin import start # start("sample.csv") start('renka_product_url_fk_snapdeal.csv')