Skip to content

Commit e1789dd

Browse files
author
agrandiere
committed
Merge branch 'dev'
2 parents 72c30e9 + 539a1c9 commit e1789dd

File tree

6 files changed

+16
-8
lines changed

6 files changed

+16
-8
lines changed

examples/all.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
####### check image
66

7-
checkAll = client.check('nudity', 'wad', 'properties', 'face', 'type')
7+
checkAll = client.check('nudity', 'wad', 'properties', 'faces', 'type')
88

99
output = checkAll.set_file('/path/to/local/file.jpg')
1010
output2 = checkAll.set_url('https://d3m9459r9kwism.cloudfront.net/img/examples/example7.jpg')

examples/nudity.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,5 +26,7 @@
2626

2727
####### check video
2828

29-
checkNudity = client.check('nudity')
30-
checkNudity.video('http://www.quirksmode.org/html5/videos/big_buck_bunny.webm', 'http://requestb.in/1nm1vw11')
29+
check = client.check('nudity', 'wad')
30+
output = check.video('http://www.quirksmode.org/html5/videos/big_buck_bunny.webm', 'http://requestb.in/1nm1vw11')
31+
32+
print(output)

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
setup(
44
name = 'sightengine',
55
packages = ['sightengine'],
6-
version = '1.2.1',
6+
version = '1.2.2',
77
description = 'Sightengine Python client',
88
author = 'Sightengine',
99
author_email='[email protected]',

sightengine/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = '1.2.1'
1+
__version__ = '1.2.2'

sightengine/check.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,8 +64,7 @@ def set_bytes(self, binaryImage):
6464
return output
6565

6666
def video(self, videoUrl, callbackUrl):
67-
url = self.endpoint + 'video/moderation.json?stream_url=' + videoUrl + '&callback_url=' + callbackUrl + '&api_user=' + self.api_user + '&api_secret=' + self.api_secret
68-
r = requests.get(url, headers=headers)
67+
r = requests.get(self.endpoint + 'video/check.json', params={'models': self.modelsType, 'callback_url': callbackUrl, 'stream_url': videoUrl, 'api_user': self.api_user, 'api_secret': self.api_secret}, headers=headers)
6968

7069
output = json.loads(r.text)
7170
return output

tests.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ def test_nudityModel(self):
2222

2323
def test_allModel(self):
2424
client = SightengineClient('1234', 'test')
25-
checkAll = client.check('nudity','wad','properties','type','face','celebrities')
25+
checkAll = client.check('nudity','wad','properties','type','faces','celebrities')
2626

2727
image = os.path.join(os.path.dirname(__file__), 'assets', 'image.jpg')
2828

@@ -56,5 +56,12 @@ def test_feedback(self):
5656
self.assertEqual('failure', feedback4['status'])
5757
self.assertEqual('argument_error', feedback4['error']['type'])
5858

59+
def test_video(self):
60+
client = SightengineClient('1234', 'test')
61+
check = client.check('wad','nudity','properties','type','faces','celebrities')
62+
63+
video_output = check.video('https://sightengine.com/assets/stream/examples/funfair.mp4', 'http://requestb.in/1nm1vw11')
64+
self.assertEqual('success', video_output['status'])
65+
5966
if __name__ == '__main__':
6067
unittest.main()

0 commit comments

Comments
 (0)