Python Workflow.store_data方法代码示例

# 需要导入模块: from workflow import Workflow [as 别名]
# 或者: from workflow.Workflow import store_data [as 别名]
class RatesCurrencyTest(unittest.TestCase):
    def setUp(self):
        self.wf = Workflow()
        self.wf.clear_settings()
        self.wf.clear_data()
        self.wf.clear_cache()
        rates.log = self.wf.logger

    def tearDown(self):
        pass

    def testLoadCurrencyInfo(self):
        currency_info = rates.get_currencies()

        # Checks if all itens have all info that is used by the script
        for currency, info in currency_info.iteritems():
            self.assertIn('Id', info, 'No ID for currency {}'.format(info))
            self.assertTrue(info['Id'], 'None ID specified for currency {}'.format(info))
            self.assertIn('Name', info, 'No Name for currency {}'.format(info))
            self.assertTrue(info['Name'], 'No Name for currency {}'.format(info))
            self.assertIn('Code', info, 'No Code for currency {}'.format(info))
            self.assertTrue(info['Code'], 'No Code for currency {}'.format(info))
            self.assertIn('Simbol', info, 'No Simbol for currency {}'.format(info))
            self.assertTrue(info['Simbol'], 'No Simbol for currency {}'.format(info))
            self.assertIn('Country', info, 'No Country for currency {}'.format(info))
            self.assertTrue(info['Country'], 'No Country for currency {}'.format(info))
            self.assertIn('Flag', info, 'No Flag for currency {}'.format(info))

    def test_is_float(self):
        tests = [(1, True),
                 ('asd', False),
                 (1.5, True),
                 ('1', True),
                 ('1', True)]

        for test in tests:
            self.assertEqual(rates.is_float(test[0]), test[1])

    def test_validate_currencies(self):
        currencies = rates.get_currencies()
        self.assertTrue(rates.validate_currencies([], 'BRL', 'USD', currencies, self.wf))
        self.assertFalse(rates.validate_currencies([], 'BRL', 'USDD', currencies, self.wf))
        self.assertFalse(rates.validate_currencies([], 'BRLL', 'USD', currencies, self.wf))

    def test_clear_caches(self):
        self.wf.cache_data('test_cache', 'testing cache')
        self.wf.store_data('test_store', 'testing store')
        with patch.object(sys, 'argv', 'program --clear'.split()):
            rates.main(self.wf)
        self.assertEqual(len(self.wf._items), 1)
        self.assertEqual(self.wf._items[0].title, 'Caches cleared!')
        self.assertIsNone(self.wf.stored_data('test_store'))
        self.assertIsNone(self.wf.cached_data('test_cache'))

    def test_evaluate_math(self):

        tests = [
            (['100*3', '100'], ['300', '100']),
            (['135.3*2', '234.5-5'], ['270.6', '229.5']),
            (['123/2', '61.5*50'], ['61.5', '3075.0']),
            (['123.5*2.5'], ['308.75']),
            # (['123', '/2', '61.5*50'], ['61.5', '3075.0']),
            # (['123', '/', '2', '61.5*50'], ['61.5', '3075.0']),
            # (['123', '/2', '/', '2', '61.5*50'], ['30.75', '3075.0']),
            # (['123', '/2', '/', '2', '61.5*50', '/3', '*2'], ['30.75', '2050.0'])
            # (['123*', '2'], ['246'])
            # (['100*2', 'usd', 'brl'], ['200', 'usd', 'brl'])
        ]

        for t in tests:
            result = rates.evaluate_math(t[0])
            self.assertEqual(result, t[1])

    def test_fmt_number(self):
        tests = [
            ('100.00', '.', '100.00'),
            ('100.00', ',', '100,00'),
            ('1.000,00', '.', '1,000.00'),
            ('100', None, '100')
        ]

        for t in tests:
            result = rates.fmt_number(t[0], t[1])
            self.assertEqual(result, t[2])

# 需要导入模块: from workflow import Workflow [as 别名]
# 或者: from workflow.Workflow import store_data [as 别名]
class TestEbook(unittest.TestCase):
    def setUp(self):
        self.wf = Workflow()
        self.wf.reset()
        itebooks.log = self.wf.logger

    def tearDown(self):
        self.wf.reset()

    def test_copy_download_link(self):
        with patch.object(sys, 'argv', 'program --copy-download-link 1529159300'.split()):
            ret = itebooks.main(self.wf)
        self.assertEqual(ret, 0)

    def test_open_in_browser(self):
        with patch.object(sys, 'argv', 'program --open-ebook-browser 1529159300'.split()):
            ret = itebooks.main(self.wf)
        self.assertEqual(ret, 0)
        self.assertEqual(len(self.wf._items), 0)

    def test_search_many_words(self):
        with patch.object(sys, 'argv', 'program Elastic Search'.split()):
            ret = itebooks.main(self.wf)
        self.assertEqual(ret, 0)
        self.assertGreater(len(self.wf._items), 0)

    def test_search_with_no_result(self):
        with patch.object(sys, 'argv', 'program Zupao in the mountains'.split()):
            ret = itebooks.main(self.wf)
        self.assertEqual(ret, 1)
        self.assertEqual(len(self.wf._items), 1)

        return_msg = self.wf._items[0].title

        self.assertTrue(return_msg.find('No books found for the query') != -1,
                        'Should return No books found for the query, returned {}'.format(return_msg))

    def test_not_enough_query_character(self):
        with patch.object(sys, 'argv', 'program ja'.split()):
            ret = itebooks.main(self.wf)

        ret_items = self.wf._items
        ret_expected_msg = 'Please, type at least {} characters to start the search...'.format(itebooks.min_characters)

        self.assertTrue(ret, 1)
        self.assertTrue(len(ret_items), 1)
        self.assertTrue(ret_items[0].title == ret_expected_msg)

    def test_show_download_progress(self):
        downloading_books = {}

        books = [('1234', 'Effective Java', 'DOWNLOADING'),
                 ('1235', 'Effective Akka', 'FINISHED'),
                 ('1236', 'Effective POG', 'DOWNLOADING'),
                 ('1237', 'Effective O Primo Basilio', 'DOWNLOADING'),
                 ('1238', 'Os caras que estao falando', 'FINISHED')]

        for book in books:
            downloading_books[book[0]] = {'title': book[0],
                                          'book_id': book[1],
                                          'status': book[2]}

        self.wf.store_data(download.STORED_DOWNLOADING_BOOKS, downloading_books)

        with patch.object(sys, 'argv', 'program --status-download'.split()):
            ret = itebooks.main(self.wf)

        ret_items = self.wf._items

        self.assertEqual(ret, 0)
        self.assertEqual(len(ret_items), 5)

    def test_show_download_progress_no_results(self):
        with patch.object(sys, 'argv', 'program --status-download'.split()):
            ret = itebooks.main(self.wf)

        ret_items = self.wf._items

        self.assertEqual(ret, 1)
        self.assertEqual(len(ret_items), 1)
        self.assertEqual(ret_items[0].title, 'No downloads running at moment.')

# 需要导入模块: from workflow import Workflow [as 别名]
# 或者: from workflow.Workflow import store_data [as 别名]
#!/usr/bin/python
# encoding: utf-8

import sys
from workflow import Workflow, web
from collections import Counter
from workflow.background import run_in_background, is_running


wf = Workflow()



arg = wf.args[0]
#arg = wf.args[0]


if arg.startswith("Username:"):
    username = arg.split(":")[1]
    wf.store_data("Username", username)
    arg = "Saved " + arg
elif arg.startswith("Token:"):
    token = arg.split(":")[1]
    wf.save_password('Token',token)

wf.send_feedback()

# 需要导入模块: from workflow import Workflow [as 别名]
# 或者: from workflow.Workflow import store_data [as 别名]
#!/usr/bin/python
# encoding: utf-8

import sys
from workflow import Workflow

file_loc = ' '.join(sys.argv[1:])
wf = Workflow()
wf.store_data('codebox_location', file_loc)

 

# 需要导入模块: from workflow import Workflow [as 别名]
# 或者: from workflow.Workflow import store_data [as 别名]
TABLENAME = "unit_alfred_workflow"
INIT_TABLE = {("MB", "B"): 1024 * 1024,
              ("MB", "KB"): 1024,
              ("MB", "GB"): 1/1024,
              ("B", "KB"): 1/1024,
              ("B", "MB"): 1/1024 * 1/1024,
              ("B", "GB"): 1/1024 * 1/1024 * 1/1024,
              ("KB", "B"): 1024,
              ("KB", "MB"): 1/1024,
              ("KB", "GB"): 1/1024 * 1/1024,
              ("GB", "B"): 1/1024 * 1/1024 * 1/1024,
              ("GB", "KB"): 1/1024 * 1/1024,
              ("GB", "MB"): 1/1024}
wf = Workflow()
if wf.stored_data(TABLENAME) is None:
    wf.store_data(TABLENAME, INIT_TABLE)
TABLE = wf.stored_data(TABLENAME)


ACTION_PATTERN = r'^(?P<number>\d+(\.\d+)*)(?P<type>[a-zA-Z]+)$'
def next_action(query):
    matched = re.match(ACTION_PATTERN, query)
    if matched is None:
        return 0, "***"

    typo = matched.group("type").upper()
    if typo in [f for (f, _) in TABLE.keys()]:
        return float(matched.group("number")), typo

    return float(matched.group("number")), "***"

# 需要导入模块: from workflow import Workflow [as 别名]
# 或者: from workflow.Workflow import store_data [as 别名]
#!/usr/bin/python
# encoding: utf-8

import sys
from workflow import Workflow

file_loc = ' '.join(sys.argv[1:])
wf = Workflow()
wf.store_data('library_location', file_loc)

print file_loc

# 需要导入模块: from workflow import Workflow [as 别名]
# 或者: from workflow.Workflow import store_data [as 别名]
    #Check if the user wants to initiate settings saving
    if wf.pargs.reset:
        wf.delete_password(u'yt_password')
        wf.reset()
        sys.exit()

    if wf.pargs.set:
        params = split_query_to_params(wf.pargs.set)
        if len(params) == 3:
            # if we have 3 params, means the user already chose a setting and filled it
            log.error('User wants to save something!! %s' % wf.pargs)
            if params[0] == 'yt_password':
                wf.save_password(u'yt_password', params[1])
            else:
                wf.store_data(params[0], params[1])
            print "Saved %s sucesfully!" % yt_title(params[0])
            sys.exit()
        if params[1] == '':
            wf.add_item('Please enter your YouTrack %s' % yt_title(params[0]),subtitle=u'Cannot be empty!', icon=ICON_SETTINGS)
        else:
            wf.add_item('Set your YouTrack %s to \'%s\'' % (yt_title(params[0]),params[1]),
                        subtitle=u'Hit enter to set.',
                        icon=ICON_SETTINGS,
                        arg=wf.pargs.set + SEPARATOR,
                        valid=True)
        wf.send_feedback()

    settings = {}
    settings['yt_url'] = wf.stored_data('yt_url')
    settings['yt_username'] = wf.stored_data('yt_username')